一.准备工作
1.系统环境是ubuntu
2.去anaconda官网下载anaconda3,python版本是3.5以上的就行,官网最新的是python3.7,没有关系,反正自己可以创建新环境
来选择其他版本的python.
3.去清华镜像源下载对应版本的pytorch,这里作者使用的是pytorch-0.4.1-py36_cuda9.0.176_cudnn7.1.2_1.tar.bz2
这个文件名意思是:python3.6版本,cuda9.0,cudnn7.1.(旧版本的pytorch的函数名与新版本的不匹配,尽量选择0.3以上
版本pytorch)
二.安装
1.首先安装anaconda,具体安装方式网上查找,添加环境变量,这样才能在控制端输入"conda"指令.
2.创建环境"mypytorch":
conda create -n mypytorch python=3.6
3.进入环境
source activate mypytorch
4.然后在mypytorch环境中利用conda命令安装下载好的pytorch文件
conda install /home/user/pytorch-0.4.1-py36_cuda9.0.176_cudnn7.1.2_1.tar.bz2
5.安装完pytorch后,在继续conda 安装pytorch,为什么安装两遍呢,因为如果直接利用conda安装,会从官网下载安装,速度太慢
(你翻墙也行,速度快可以忽略这个方法),第二次是利用pytorch官网给的命令,当conda发现本地已经装了pytorch后,他不会再
安装了,但是会把pytorch需要的其他依赖包给装好,比如cuda9.0等,这些包是从系统设置的源中下载的(比如aliyun)
conda install pytorch torchvision -c pytorch
6.这样pytorch就安装完了