Pycharm中import torch报错解决方案

2024-03-19 2529阅读

1.出现原因

一种最常见的安装pytorch时见到的问题,基本上不外乎是以下三个原因。

Pycharm中import torch报错解决方案 第1张

原因1:pytorch,cuda,python版本不兼容

原因2:未正确创建虚拟环境或环境内未正确下载pytorch包

原因3:pycharm内没有选择对应环境(解释器)导致找不到包

2.版本不兼容

先按照下图判断cuda,python和torch版本是否兼容,cu113表示cuda11.3,cp310表示python3.10,按照对应关系判断是否兼容。

Pycharm中import torch报错解决方案 第2张 查询cuda版本:按下键盘上的win+r,输入cmd打开控制台,再输入nvcc -V(注意V是大写,中间有个空格),箭头所指出即为cuda的版本。

Pycharm中import torch报错解决方案 第3张

查询python版本:按下键盘上的win+r,输入cmd打开控制台,再输入python,箭头所指即为python的版本。

Pycharm中import torch报错解决方案 第4张

 查询pytorch版本:首先打开Anaconda Prompt,输入conda activate xxxx进入你创建的虚拟环境,xxxx是你虚拟环境的名字,再输入conda list就可以看到该环境下有的所有软件包,找到pytorch,箭头所指即为pytorch版本。

Pycharm中import torch报错解决方案 第5张

Pycharm中import torch报错解决方案 第6张

Pycharm中import torch报错解决方案 第7张

完成上述工作,将不兼容的cuda,python或者pytorch重新下载即可,这三项的下载安装方法在此不赘述。

3.检查虚拟环境和torch包

Anaconda Prompt基本指令:

conda create –n 虚拟环境名字 python=版本    (创建环境指令)

conda env list                                                      (查看虚拟环境列表)

conda activate 虚拟环境名字                              (激活/进入某个虚拟环境)

conda list                                                          (查看当前虚拟环境下的软件包)

检查环境:进入虚拟环境下,输入python进入控制台,输入import torch和torch.cuda.is_available()代码,若返回true说明虚拟环境是存在的并且环境内已经正确下载了pytorch软件包,问题是出在pycharm的配置上。

Pycharm中import torch报错解决方案 第8张

3.Pycharm内配置环境/解释器

题主就是在这里卡了很久,终于发现是解释器没有配置好,需要自己选择conda环境和配置解释器。

右上角齿轮打开进入设置

Pycharm中import torch报错解决方案 第9张

按照顺序点击,选择添加本地解释器。

Pycharm中import torch报错解决方案 第10张

选择conda环境,使用现有环境,conda执行文件找到Anaconda安装路径下Scripts文件夹内的conda.exe,最后选择含有torch软件包的虚拟环境,题主创建名为pytorch。

Pycharm中import torch报错解决方案 第11张

创建完解释器后,下方会显示出该解释器/虚拟环境下的所有软件包,看到有pytorch包即选择了正确的环境和解释器,点击确定和应用即完成了pycharm内的pytorch配置并选择了正确的虚拟环境。

Pycharm中import torch报错解决方案 第12张

创建一个.py文件并输入以下代码检测是否成功连接

Pycharm中import torch报错解决方案 第13张

成功导入

Pycharm中import torch报错解决方案 第14张

❗❗特别地

有时在选择环境创建本地解释器时会提示报错  请指定不同的SDK名称 并且拒绝选择和应用解释器,此时选择下拉页面,选择全部显示。

Pycharm中import torch报错解决方案 第15张

点筛选器后,删除掉所有飘红,带叉和重复路径的解释器到没有红色和报错,即解决问题。

Pycharm中import torch报错解决方案 第16张

至此通过以上三个方面排查并解决pycharm中无法导入pytorch的问题。


    免责声明:我们致力于保护作者版权,注重分享,被刊用文章因无法核实真实出处,未能及时与作者取得联系,或有版权异议的,请联系管理员,我们会立即处理! 部分文章是来自自研大数据AI进行生成,内容摘自(百度百科,百度知道,头条百科,中国民法典,刑法,牛津词典,新华词典,汉语词典,国家院校,科普平台)等数据,内容仅供学习参考,不准确地方联系删除处理! 图片声明:本站部分配图来自人工智能系统AI生成,觅知网授权图片,PxHere摄影无版权图库和百度,360,搜狗等多加搜索引擎自动关键词搜索配图,如有侵权的图片,请第一时间联系我们,邮箱:ciyunidc@ciyunshuju.com。本站只作为美观性配图使用,无任何非法侵犯第三方意图,一切解释权归图片著作权方,本站不承担任何责任。如有恶意碰瓷者,必当奉陪到底严惩不贷!

    目录[+]