ubuntu16.04下用miniconda安装pytorch GPU环境记录

1.安装miniconda

进入Miniconda官网Miniconda下载相应版本
ubuntu16.04下用miniconda安装pytorch GPU环境记录_第1张图片
我下载的是Python3.7 64bit版本
下载完成后,cd进入下载目录,执行

sh YOUR_FILE_NAME -b

sh Miniconda3-latest-Linux-x86_64.sh -b

然后执行

~/miniconda3/bin/conda init

此时重新启动终端,执行

conda -V

如果不报错则安装成功
ubuntu16.04下用miniconda安装pytorch GPU环境记录_第2张图片

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/

然后执行它,片刻之后即可安装成功
ubuntu16.04下用miniconda安装pytorch GPU环境记录_第3张图片

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"

如果全部都未报错,那么就一切安装成功!
ubuntu16.04下用miniconda安装pytorch GPU环境记录_第4张图片

你可能感兴趣的:(cuda,anaconda,ubuntu)