本文主要是介绍解决AssertionError: Torch not compiled with CUDA enabled，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
运行环境：win10+cdua11.0+python3.8+pytorch2.0.1
报错语句：weights = torch.Tensor(TRAIN_DATASET.labelweights).cuda()
报错信息：
Exception has occurred: AssertionError
Torch not compiled with CUDA enabled
原因：安装的pytorch与cudnn的版本不匹配
解决方法：
其中一种方法是重新安装pytorch
pip install torch==1.7.1+cu110 -f https://download.pytorch.org/whl/torch_stable.html如果您的电脑上安装的是其它版本的cuda，那么可以查看下面的可下载安装的pytorch的版本：
versions: 1.4.0, 1.4.0+cpu, 1.4.0+cu92, 1.5.0, 1.5.0+cpu, 1.5.0+cu101, 1.5.0+cu92, 1.5.1, 1.5.1+cpu, 1.5.1+cu101, 1.5.1+cu92, 1.6.0, 1.6.0+cpu, 1.6.0+cu101, 1.7.0, 1.7.0+cpu, 1.7.0+cu101, 1.7.0+cu110, 1.7.1, 1.7.1+cpu, 1.7.1+cu101, 1.7.1+cu110, 1.8.0, 1.8.0+cpu, 1.8.0+cu101, 1.8.0+cu111, 1.8.1, 1.8.1+cpu, 1.8.1+cu101, 1.8.1+cu102, 1.8.1+cu111, 1.9.0, 1.9.0+cpu, 1.9.0+cu102, 1.9.0+cu111, 1.9.1, 1.9.1+cpu, 1.9.1+cu102, 1.9.1+cu111, 1.10.0, 1.10.0+cpu, 1.10.0+cu102, 1.10.0+cu111, 1.10.0+cu113, 1.10.1, 1.10.1+cpu, 1.10.1+cu102, 1.10.1+cu111, 1.10.1+cu113, 1.10.2, 1.10.2+cpu, 1.10.2+cu102, 1.10.2+cu111, 1.10.2+cu113, 1.11.0, 1.11.0+cpu, 1.11.0+cu113, 1.11.0+cu115, 1.12.0, 1.12.0+cpu, 1.12.0+cu113, 1.12.0+cu116, 1.12.1, 1.12.1+cpu, 1.12.1+cu113, 1.12.1+cu116, 1.13.0, 1.13.0+cpu, 1.13.0+cu116, 1.13.0+cu117, 1.13.1, 1.13.1+cpu, 1.13.1+cu116, 1.13.1+cu117, 2.0.0, 2.0.0
+cpu, 2.0.0+cu117, 2.0.0+cu118, 2.0.1, 2.0.1+cpu, 2.0.1+cu117, 2.0.1+cu118, 2.1.0, 2.1.0+cpu, 2.1.0+cu118, 2.1.0+cu121
pytorch的cu版本文件很大，一般有3G多，安装比较费时，网速快的情况下，可以较快安装好，安装好之后，再次运行程序，可以正常运行了，大功告成！
这篇关于解决AssertionError: Torch not compiled with CUDA enabled的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
