1.安装miniconda
进入Miniconda官网Miniconda下载相应版本
我下载的是Python3.7 64bit
版本
下载完成后,cd
进入下载目录,执行
sh YOUR_FILE_NAME -b
如sh Miniconda3-latest-Linux-x86_64.sh -b
然后执行
~/miniconda3/bin/conda init
此时重新启动终端,执行
conda -V
如果不报错则安装成功
2.配置pytorch环境
首先新建一个环境,如环境名设为pytorch,python版本设为3.7
conda create -n pytorch python=3.7
环境创建完毕后,重启终端,执行
conda activate pytorch
接下来就是pytorch及CUDA的安装
由于通过Pytorch官网进行安装,网络会极慢。所以需要设置清华镜像源,进行满速下载。
依次执行如下命令:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
接着开始正式安装:
首先进入pytorch官网选取合适的版本组合,如果该页面提供的版本都不合适,可进入pytorch历史版本查找合适的版本。
这里我选择了pytorch1.0.0+torchvision0.2.1+cuda8.0版本,并找到对应的如下安装命令:
conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch
但是此时还需要将指令末的pytorch替换为清华源,如下:
conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
然后执行它,片刻之后即可安装成功
3.验证安装是否成功
在该环境中,依次执行如下命令
python
import torch # 如正常则静默
a = torch.Tensor([1.]) # 如正常则静默
a.cuda() # 如正常则返回"tensor([ 1.], device='cuda:0')"
from torch.backends import cudnn # 如正常则静默
cudnn.is_acceptable(a.cuda()) # 如正常则返回 "True"
如果全部都未报错,那么就一切安装成功!