原文链接:Anaconda+Pycharm环境下的PyTorch配置方法
问题:
torch包在pycharm里面的导入问题
描述: 在cmd端 import torch 没问题,在pycharm里导入的时候不成功。运行代码的时候会报错:
import torch
ImportError: No module named 'torch'
本文Python版本为3.5。存在python版本和torch版本不一致的问题。
解决方法:
在cmd端创建新的conda
虚拟环境
(官方文档
https://conda.io/projects/conda/en/latest/user-guide/tasks/manage-environments.html)
step1: 输入:
conda create --name pytorch-py36 --channel pytorch python=3.6 pytorch-cpu torchvision-cpu
activate pytorch-py36
python
step3: 在打开的python
命令行里输入
import torch
import torchvision
import numpy
不报错则说明构建conda
虚拟环境pytorch-py36
成功。
注意:pytorch-py36
默认在anaconda
目录下的envs
文件夹中。
将自己创建的conda
虚拟环境pytorch-py36
添加到pycharm
的interpreter
可选列表中。
A: 在pycharm里
新建一个project.
- 新建工程界面,选择
Existing enviroment
,点击...
按钮,会弹出添加界面。 settings
界面,点击齿轮
按钮,选择Add
,会弹出添加界面。
在添加界面左侧选择Conda Environment
,右边选择Existing environment
,再点击...
按钮添加一个已经存在的conda
虚拟环境,即刚刚构建的pytorch-py36
。
B: 在interpreter
列表中选择刚刚添加的interpreter
。
如果需要对刚添加的interpreter
改名,比如把python 3.6 (3)
改成pytorch-py36
,则点击齿轮
按钮,点击show all
,里面可以改名。
试着使用import torch
,import torchvision
这样的语句,不会报错则成功。