pytorch 在ubuntu16.04 下安装

来源:互联网 发布:手机网络无法登陆钉钉 编辑:程序博客网 时间:2024/06/06 01:31

电脑环境:

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'


原创粉丝点击