前言:
我目前已有的配置: Windows10 + Anaconda3 + Python 3.8 + CUDA10.2 + cuDNN
最近在跑一个人眼视线的模型,需要import torch, torchvision. 我之前已经在pytorch环境中安装了pytorch 和torchvision了,打开Anaconda prompt, 在python中可以import torch 和 torchvision,但是在pycharm中始终无法import. 后面寻找原因,发现了两个问题:
- 我的CUDA, cudnn, pytorch , torchvision, python安装的版本都不匹配(需要安装教程可以参考我之前的博客)
- 在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
问题得到解决。
参考链接:
- https://www.94e.cn/info/1783
- https://blog.csdn.net/weixin_45838363/article/details/104342104
- 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
- https://blog.csdn.net/weixin_39450145/article/details/104801730(安装cv2)