pytorch 在ubuntu16.04 下安装

前端之家收集整理的这篇文章主要介绍了pytorch 在ubuntu16.04 下安装前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

电脑环境:

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'

猜你在找的Ubuntu相关文章