电脑环境:
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'