ubuntu上安装pytorch-gpu(前提是gpu驱动都配置好)

1.在Anaconda官网下载

https://repo.anaconda.com/archive/Anaconda3-2019.03-Linux-x86_64.sh

安装

bash Anaconda3-2019.03-Linux-x86_64.sh

修改环境变量

vim ~/.bashrc
添加 export  PATH="/root/anaconda3/bin:$PATH"
source ~/.bashrc

验证；安装完成之后，打开终端，输入python，出现如下信息；

2.安装PyTorch

针对cuda8.0

conda install pytorch=0.4.1 cuda80 -c pytorch

具体可参考

https://pytorch.org/get-started/previous-versions/

验证是否安装成功

import torch
x= torch.Tensor([1.0])
xx= x.cuda()
print(xx)

结果如下所示