Python&aconda系列:Torch not compiled with CUDA enabled 报错的归纳总结
    		       		warning:
    		            这篇文章距离上次修改已过433天,其中的内容可能已经有所变动。
    		        
        		                
                报错解释:
这个错误表明你正在尝试使用PyTorch库进行深度学习计算,但是PyTorch没有被编译为支持CUDA(一种用于图形处理的NVIDIA技术)的版本。CUDA是在显卡上进行并行计算的一种技术,对于使用NVIDIA显卡的用户来说,拥有CUDA支持的PyTorch可以显著提升计算性能。
问题解决方法:
- 确认你的电脑中安装了支持CUDA的NVIDIA显卡。
- 安装支持CUDA的PyTorch版本。可以通过Anaconda的命令行界面安装正确的版本。例如,如果你想要安装支持CUDA 10.2的PyTorch版本,可以使用以下命令: - conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch- 注意:请根据你的CUDA版本调整 - cudatoolkit=10.2中的版本号。
- 如果你不确定你的显卡支持CUDA,或者你不希望使用CUDA,可以安装一个不需要CUDA支持的PyTorch版本: - conda install pytorch torchvision torchaudio cpuonly -c pytorch
- 如果你已经安装了不支持CUDA的PyTorch,但是尝试使用CUDA的功能,那么你需要更新你的代码,确保不使用CUDA特有的功能或对象。例如,如果你的代码中有类似tensor.cuda()的语句,你需要将它们改为使用tensor.to('cuda'),这样即使在不支持CUDA的环境中也能正常运行。
评论已关闭