原因
原因是因为自己下的对应torch的cuda和计算机本身的cuda版本不一致,导致了Torch not compiled with CUDA enabled,举例来说就是假设计算机本身更新的cuda是12.1的,但是可能由于之前下载pytorch对应的cuda是11.6的,这时候在运行时就会报错。
解决方法
让计算机的cuda和pytorch的cuda版本对应一致。按如下步骤进行:
(1):进入anaconda里自己的的深度学习虚拟环境,conda active xxx,conda list查看pytorch对应的cuda。

可以看到torch版本对应的是2.3.1 ,cuda是12.1.
(2):看自己NVIDIA能接受cuda的最高型号是多少(后续下载要小于最高型号):第一步点击系统信息,第二部点击组件,以此电脑为例,可以看到允许的最高型号是12.2


(3):了解型号之后,去pytorch对应官网下载:复制’Run this Command‘的代码,去anaconda里下载即可。

(4):再次运行代码。
3080

被折叠的 条评论
为什么被折叠?



