windows7 安装pytorch

这几天为了运行python的图像转换的项目,不得不安装pytorch,安装了两天,最后把经验记录一下。python

若是版本不匹配会抛出不少错误,而网上的各类解决方式有大部分也解决不了问题。spa

在安装pytorch以前,首先是你的PC安装的是NVIDIA显卡,而后打开NVIDIA控制面板,查看CUDA的版本,老的显卡建议就到8.0,由于9.0的cuda要求显卡的计算能力在3.5以上.net

个人显卡计算能力只有3.0。若是显卡的版本较低须要进行cuda版本升级。3d

运算能力参照:https://blog.csdn.net/real_myth/article/details/44308169blog

NVIDIA版本下载:https://developer.nvidia.com/cuda-toolkit-archiveit

显卡版本安装以后,须要安装Anaconda3,参照网上的例子若是这时候进行pytorch安装,会默认安装cuda9.1,即便你指定cuda80也很差用,因此只能下载离线版本的class

cuda为80版本的pytorch。终端

https://pan.baidu.com/s/1dF6ayLr?errno=0&errmsg=Auth%20Login%20Sucess&&bduss=&ssnerror=0&traceid=#list/path=%2Fpytorch下载

下载到本地以后,在Anaconda Prompt终端运行:numpy

 

conda install numpy mkl cffi

conda install --offline pytorch-0.2.1-py36he6bf560_0.2.1cu80.tar.bz2 #离线安装

完了!

相关文章
相关标签/搜索