今日在跑代码过程中,出现如下错误:
Torch not compiled with CUDA enabled
首先,查看CUDA的版本,在cmd中输入
nvidia-smi
右上角 CUDA Version显示我的版本是11.0
我倾向于不改变CUDA的版本,但是主页并没有CUDA 11.0的选项(pytorch主页:PyTorch)
点击下方的Previous Versions Of Pytorch,即https://pytorch.org/get-started/previous-versions/
此页面展示了不同版本pytorch+不同版本cuda的安装语句。
我的CUDA 11.0对应的版本是pytorch 1.7.1
如图,conda里的安装语句为(在conda中类似于cmd的页面运行,一次可能不成功,可多运行几次)
# CUDA 11.0
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
如图,windows中的运行语句为(可在cmd中运行,一次可能不成功,多安装几次)
# CUDA 11.0
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
如果你的电脑中存在多版本的Python,安装语句则变为
cd C:\Users\(用户名)\AppData\Local\Programs\Python\Python38
python38 -m pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
这里使用python38是因为
我将C:\Users\(用户名)\AppData\Local\Programs\Python\Python38(下载后自带我未修改)中
下面的文件重命名为python38(我仅修改了该文件的名称)都配置好后再次运行代码,出现新的报错:
module 'torch' has no attribute 'nan'
查完资料发现tytorch1.8以上采用这个功能,好了我不得不去更新CUDA版本了,
安装CUDA,指路windows安装cuda11.1、cudnn及pytorch_cuda11.1对应的cudnn_烫烫烫烫烫烫丶的博客-CSDN博客我安装了CUDA11.7,并安装了最新版本的torch2.0.1,仍旧报错,
然后安装了torch1.13.1顺利解决问题,
安装哪个版本因程序而已,可以自己多试试