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包

  1. 打开cmd窗口,用CD命令,将当前目录定位到anaconda3目录下(我安装的是anaconda3),再用dir命令查看anaconda3中的文件夹,你会发现anaconda3文件夹下面有Scripts这个文件夹的
    Pycharm中import torch报错,python中import torch不报错_第1张图片
    Pycharm中import torch报错,python中import torch不报错_第2张图片
  2. 用cd命令,切换到Scripts文件夹中,然后我们就可以使用pip来进行安装了(pip是ananconda的pip工具),输入命令 pip install torchpip install torchvision, 然后等待安装完成即可。因为我的模型中后续还需要import cv2, 所以我顺便把cv2的包也安装完了,输入pip install opencv-python, 完成了opencv的安装。
    Pycharm中import torch报错,python中import torch不报错_第3张图片
    Pycharm中import torch报错,python中import torch不报错_第4张图片

二、 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

问题得到解决。
Pycharm中import torch报错,python中import torch不报错_第5张图片

参考链接:

  1. https://www.94e.cn/info/1783
  2. https://blog.csdn.net/weixin_45838363/article/details/104342104
  3. https://blog.csdn.net/qq_29883591/article/details/78077244?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-2.control&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-2.control
  4. https://blog.csdn.net/weixin_39450145/article/details/104801730(安装cv2)

你可能感兴趣的:(pytorch安装,opencv,pycharm)