您的位置:首页 > 其它

pytorch 在ubuntu16.04 下安装

2017-11-09 21:54 1076 查看
电脑环境:

Python 2.7.13 |Anaconda

CUDA 8.0

显卡:GTX1060 6G

欲安装版本:pytorch 0.2.0_3

pytorch官网有安装教程,我按照官网的指示进行安装,下载whl包文件太慢。于是就自己回到宿舍下载了whl包,通过pip方式安装。

附上pytorch的安装文件百度云链接:链接: https://pan.baidu.com/s/1nuFHVSX 密码: g2eq

然后进入whl文件所在目录,在终端下输入:pip install torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whl

在import torch的时候,出现了

ImportError: numpy.core.multiarray failed to import 错误

这个可能是因为numpy版本不对,重新安装并更新numpy包就行了

pip install -U numpy -i https://mirrors.aliyun.com/pypi/simple/
在后面加上-i参数,添加阿里云的源或则其他源速度会很快

至此,pytorch就安装成功了

>>> import torch

>>> torch.__version__

'0.2.0_3'
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: