python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Pycharm import torch报错,python import torch不报错

Pycharm中import torch报错,python中import torch不报错的解决

作者:困坤的小菜鼠

这篇文章主要介绍了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. 后面寻找原因

发现了两个问题:

第二个原因应该是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 torchpip 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

问题得到解决。

总结

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

您可能感兴趣的文章:
阅读全文