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

问题得到解决。

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

最新评论