Pycharm中import torch报错,python中import torch不报错的解决
作者:困坤的小菜鼠
前言
我目前已有的配置:
Windows10 + Anaconda3 + Python 3.8 + CUDA10.2 + cuDNN
最近在跑一个人眼视线的模型,需要import torch, torchvision. 我之前已经在pytorch环境中安装了pytorch 和torchvision了,打开Anaconda prompt, 在python中可以import torch 和 torchvision,但是在pycharm中始终无法import. 后面寻找原因
发现了两个问题:
- 1.我的CUDA, cudnn, pytorch , torchvision, python安装的版本都不匹配
- 2.在Pycharm中import torch报错,原因项目的解释器路径是Python的安装路径,需要改为Anaconda的Python的路径即可,而且我之前的pytorch, torchvision并不是安装在我的anaconda中的python环境中。
第二个原因应该是pycharm无法正常import torch 和torchvision的主要原因。
为了之后的工作方便,后面我还是将这些工具全都重新安装了一遍,最后的版本为 CUAD 10.2 ---- pytorch 1.6.0 ----torchvision 0.7.0。
如果你也遇到了这个pycharm 的import的问题的话,建议你先处理第二个原因试一下
下面我记录一下我的解决办法:
一、anaconda下载pytorch, torchvision包
打开cmd窗口,用CD
命令,将当前目录定位到anaconda3目录下(我安装的是anaconda3),再用dir
命令查看anaconda3中的文件夹,你会发现anaconda3文件夹下面有Scripts这个文件夹的
用cd命令,切换到Scripts文件夹中,然后我们就可以使用pip来进行安装了(pip是ananconda的pip工具)
输入命令 pip install torch
和pip install torchvision
, 然后等待安装完成即可。
因为我的模型中后续还需要import cv2, 所以我顺便把cv2的包也安装完了
输入pip install opencv-python
, 完成了opencv的安装。
二、Pycharm中import torch, import torchvision报错
在Pycharm中import torch报错,原因项目的解释器路径是Python的安装路径,改为Anaconda的Python的路径即可
具体步骤为:
File–Settings–Project:xxx–Project Interpreter–show all–点击“+”添加–System Interpreter–C:\Users\Administrator\Anaconda3\python.exe
问题得到解决。
总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。