2021
03-29
03-29
Linux安装Pytorch1.8GPU(CUDA11.1)的实现
先说下自己之前的环境(都是Linux系统,差别不大):Centos7.6NVIDIADriverVersion440.33.01(等会需要更新驱动)CUDA10.1Pytorch1.6/1.7提示,如果想要保留之前的PyTorch1.6或1.7的环境,请不要卸载CUDA环境,可以通过Anaconda管理不同的环境,互不影响。但是需要注意你的NVIDIA驱动版本是否匹配。在这里能够看到官方给的对应CUDA版本所需使用驱动版本。通过上表可以发现,如果要使用CUDA11.1,那么需要将显卡的驱动更...
继续阅读 >