当服务器上的CUDA版本过低，导致无法安装高版本的PyTorch时，可以避免复杂的系统级升级，转而在虚拟环境中操作。通过conda创建一个名为ncuda10的虚拟环境，并激活它，然后安装CUDA工具包10.1和CUDNN 7.6.5。完成这些步骤后，就可以在该环境中安装与CUDA版本兼容的PyTorch了。
CUDA版本不够，升级又太复杂，应该怎么办呢？
服务器cuda的版本是9.0，只能安装很低版本的pytorch，查了查怎么升级cuda，感觉好复杂，幸好可以在虚拟环境中安装cuda 10.1，省了不少力气。在虚拟环境中安装新版本的cuda
1、创建虚拟环境。（注意cuda10为虚拟环境的名字）
conda create -n cuda10 python=3.7
2、进入环境后，安装cuda和cudnn
conda activate cuda10
conda install cudatoolkit=10.1
conda install cudnn=7.6.5
这样就结束啦，可以去pytorch的官网安装对应版本的torch啦！
扫一扫
1762
到【灌水乐园】发言
