Ubuntu16.04:在anaconda下安装pytorch-gpu

1 创建虚拟环境并进入

conda create -n pytorch-gpu python=3.6
conda activate pytorch-gpu

 

2 下载对应的安装包和配件

方法一(推荐)：

conda install pytorch-gpu

conda install mkl

方法二：

到清华镜像下载对应版本的pytorch包https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/

到下载好的目录下安装：

cd ~/Downloads/
conda install --offline pytorch-1.0.1-py3.6_cuda8.0.61_cudnn7.1.2_0.tar.bz2

conda install mkl

3 检验

python

import torch

print(torch.cuda.is_available())