用conda导入pytorch的问题

时间:2018-12-09 22:56:37

标签: python python-3.x graph anaconda pytorch

我的系统是:

 x86_64
DISTRIB_ID=LinuxMint
DISTRIB_RELEASE=17.1
DISTRIB_CODENAME=rebecca
DISTRIB_DESCRIPTION="Linux Mint 17.1 Rebecca"
NAME="Ubuntu"
VERSION="14.04.5 LTS, Trusty Tahr"
ID=ubuntu
ID_LIKE=debian
PRETTY_NAME="Ubuntu 14.04.5 LTS"
VERSION_ID="14.04"
HOME_URL="http://www.ubuntu.com/"
SUPPORT_URL="http://help.ubuntu.com/"
BUG_REPORT_URL="http://bugs.launchpad.net/ubuntu/"
cat: /etc/upstream-release: Is a directory

我最近安装了pytorch,因为它是使用Python因果发现工具箱所必需的,并且我已经使用Anaconda而不使用CUDA来安装了它,就像这样:

conda安装pytorch-cpu torchvision-cpu -c pytorch

此安装成功,但是当我尝试运行时

import torch

在IDLE中,找不到模块。我意识到这个问题可能是多余的,但是我搜索了类似的问题却没有找到任何可行的方法,而且我很准时。另外,我对Anaconda不熟悉,所以我不确定如何通过安装它来与其他Python一起使用。预先感谢您的帮助。

1 个答案:

答案 0 :(得分:3)

如果您想使用PyTorch,请使用

将其安装在本地环境中
conda create -n pytorch_env  python=3  
source activate pytorch_env   
conda install pytorch-cpu torchvision -c pytorch

转到python shell并使用命令导入

import torch