anaconda+pytorch_gpu安装

@anaconda+pytorch_gpu

平台环境
(1) ubuntu 16.04 64位系统
(2) 一块 TITAN Xp,驱动版本NVIDIA-Linux-x86_64-430.63.01.run
(3) cuda版本10.1,cuda_10.1.243_455.32.00_linux.run
系统及驱动、cuda安装请自行安装,本文档假定上述已安装完成

1 下载anaconda

我使用的是anaconda3_5.3.1-Linux-86_64.sh,
2 安装anaconda
./ anaconda3_5.3.1-Linux-86_64.sh

3 创建conda 的虚拟环境
conda create -n yolov5 python=3.7
4 安装pytorch

安装符合我的驱动的pytorch版本，我安装的是

pip3 install torch1.8.0+cu101 torchvis ion0.9.1+cu101 torchaudio===0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

5 查看是否安装成功

进入python

import torch
torch.cuda.is_available()

如何是true 则安装成功！