报错信息:
AssertionError:Torch not compiled with CUDA enabled
实现方案:
1、查看自己GPU版本,
DOS窗口,输入命令,nvidia-smi
查看结果,12.5
2、查看自己CUDA版本,
DOS窗口,输入命令,nvcc --version
查看结果,11.6
3、查看自己python版本,
DOS窗口,输入命令,python -V
查看结果,3.10
4、再python官网,
https://pytorch.org/get-started/previous-versions/ 查看自己的conda的文件版本
5、再wheel网站,
https://download.pytorch.org/whl/torch_stable.html 下载对应的依赖文件库
6、下载依赖库文件解释,
CUDA版本对应cu116;依赖库版本对应torchaudio-0.13.1;python版本对应cp310
cu116/torch-1.13.1%2Bcu116-cp310-cp310-win_amd64.whl
cu116/torchvision-0.14.1%2Bcu116-cp310-cp310-win_amd64.whl
cu116/torchaudio-0.13.1%2Bcu116-cp310-cp310-win_amd64.whl
7、进行依赖库下载所在目录,
CMD进入DOS窗口,创建或激活conda虚拟环境
8、查看虚拟环境下所安装的依赖库,
DOS窗口,输入命令,pip list
9、把虚拟环境下已安装的冲突依赖库卸载掉,
DOS窗口,输入命令,pip uninstall torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0
10、把新下载的wheel依赖库文件,使用命令
DOS窗口,输入命令,
pip install "cu116/torch-1.13.1%2Bcu116-cp310-cp310-win_amd64.whl"
DOS窗口,输入命令,
pip install "cu116/torchvision-0.14.1%2Bcu116-cp310-cp310-win_amd64.whl"
DOS窗口,输入命令,
pip install "cu116/torchaudio-0.13.1%2Bcu116-cp310-cp310-win_amd64.whl"
11、安装成功后,使用如下命令进行查看,
DOS窗口,输入命令,pip list