AssertionError:Torch not compiled with CUDA enabled最简单解决方案

报错信息:
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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值