成功解决:AssertionError: Torch not compiled with CUDA enabled

成功解决:AssertionError: Torch not compiled with CUDA enabled

码农世界 2024-06-06 前端 118 次浏览 0个评论

在运行pycharm项目的时候,出现了以上的报错,主要可以归结于以下两个个方面:

1、没有安装GPU版本的pytorch,只是使用清华的镜像地址下载了CPU版本的pytorch

2、安装的CUDA和安装的pytorch的版本不相互对应

我使用   pip list 来查看我在该环境下安装了哪些依赖项,发现自己的torch是CPU版本的, 

如何区分的呢?主要通过安装的环境的版本号的后面时候加上了cu,比如说下面这幅图就可以看出这里采用的CUDA版本号为11.3,所以在我的环境里面没有安装GPU版本的torch.

 找到问题后,我们就可以进行解决了。

如果自己不是CUDA版本的torch,可以使用pip uninstall torch的命令来卸载已经安装好的torch后继续下面的操作.

第一步:查看自己电脑上安装的python版本,因为python的版本对应者pytorch的版本,pytorch的版本又对应着CUDA的版本。所以先查看自己的python的版本,那么如何查看自己的电脑上的python版本呢࿱

转载请注明来自码农世界,本文标题:《成功解决:AssertionError: Torch not compiled with CUDA enabled》

百度分享代码,如果开启HTTPS请参考李洋个人博客
每一天,每一秒,你所做的决定都会改变你的人生!

发表评论

快捷回复:

评论列表 (暂无评论,118人围观)参与讨论

还没有评论,来说两句吧...

Top