Pytorch安装以及Pycharm中无法导入torch包的问题


我们在安装完Anaconda之后,去Pytorch官网 : https://pytorch.org/get-started/locally/ 找到适合自己配置的版本并复制这行命令(其中的-c pytorch命令是使用官网源下载,若较慢或设置了镜像源可以把这里删去)

在这里插入图片描述

然后问题就来了,

在进入conda控制台之后,不管三七二十一立马粘上回车,然后发现死都找不到这个包,这里实际上是装到了base这个默认的环境下,我们在安装时应该先创建一个虚拟环境

conda create --name DeepLearning python=3.10(这里我的环境名是DeepLearning,注意:这里一定要指定python版本号

然后activate DeepLearning 进入创建的虚拟环境,可以直接 conda install ,也可以配置镜像源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ #设置搜索时显示通道地址 conda config --set show_channel_urls yes

关于配置镜像源参考CSDN「小猫不会去楼兰捉虫」的这篇文章,链接:Anaconda配置国内镜像源,这里不再过多阐述

然后去Pycharm,为项目添加环境和解释器

在这里插入图片描述
在这里插入图片描述

在这里插入图片描述

这里有的常见的几个问题:

1.在conda控制台下能导入torch包,pycharm提示找不到

在这里插入图片描述

问题:包所在的环境和pycharm项目的环境不一致
解决办法:控制台能过说明安装上了,但是IDE找不到,如果你知道是哪个环境那么直接在pycharm中改Interpreter(解释器),如果不知道那么如上图片所示一步步conda create重新创建环境、activate env_name 进入环境然后下载pytorch,去pycharm配置环境和解释器
2.import torch能代码提示但是运行就报错说找不到
问题:大概率就是你创建环境的时候没加python=版本号,导致创建的环境里没有编译器,这样你创建的环境就是一个空壳子,里面虽然装了包,但是无法执行,所以也就可以解释为什么pycharm能识别torch但运行报错
解决办法:不知道直接拖个python.exe过去行不行,反正我是从头创建了一遍环境
3.在Pycharm下创建Anaconda虚拟环境,但envs目录下没有interpreter解释器问题的解决
这个问题是我在第2个问题产生的时候伴随发现的,详情建议去看看CSDN「猎户在天」的文章,原文链接:在Pycharm下创建Anaconda虚拟环境,但envs目录下没有interpreter解释器问题的解决

我们使用conda create --name new_env_test 创建完虚拟环境new_env_test之后,

conda activate new_env_test 发现这个环境激活正常,

也能使用 conda info --env看到,以为这个环境可以用了,

实际上这个环境没有python解释器,所以别人的虚拟环境下是这样的
在这里插入图片描述

而你的是这样的
在这里插入图片描述

如果想要在new_env_test中拥有bin文件夹,我们必须用
conda create -n my_env python=version_number
在创建环境时将python安装到env(和刚才相比,在末尾加了一个pyhon)
或者conda install -n my_env python=version_number将Python添加到现有环境中。

然后我发现我的pycharm就可以和cuda一起正常工作啦
import torch
print(torch.cuda.is_available())#检查torch版本和cuda版本是否匹配

在这里插入图片描述

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐