本文介绍: 在运行pycharm项目的时候,出现了AssertionError: Torch not compiled with CUDA enabled,主要可以归结于以下两个个方面:1、没有安装GPU版本的pytorch,只是使用清华的镜像地址下载了CPU版本的pytorch2、安装的CUDA和安装的pytorch的版本不相互对应
在运行pycharm项目的时候,出现了以上的报错,主要可以归结于以下两个个方面:
1、没有安装GPU版本的pytorch,只是使用清华的镜像地址下载了CPU版本的pytorch
我使用 pip list 来查看我在该环境下安装了哪些依赖项,发现自己的torch是CPU版本的,
如何区分的呢?主要通过安装的环境的版本号的后面时候加上了cu,比如说下面这幅图就可以看出这里采用的CUDA版本号为11.3,所以在我的环境里面没有安装GPU版本的torch.
声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。