Ubuntu + cuda + anaconda + cudatoolkit关系说明

Ubuntu：带有GPU的宿主机

1.通过指令查看支持的驱动，比如我的电脑支持nvidia-driver-460

# 查看支持的nvidia驱动
$ubuntu-driver devices

2.安装驱动之后查看支持的cuda版本,安装完成nvidia-driver-460使用下面指令可以看到支持的cuda版本为cuda11.2

$nvidia-smi

3.安装cuda和cudnn,我的驱动支持的cuda版本为11.2，则我可以安装11.2以内的cuda，比如我安装了cuda10.1

4.安装完成cuda10.1之后，安装anaconda创建虚拟环境，创建虚拟换的时候可以使用默认安装的系统cuda版本,也可以创建时指定一个新的cuda

# 比如
$conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

这里的cudatoolkit就是安装在虚拟环境的cuda, 这个虚拟环境的cuda版本最好不要超过系统内安装的cuda10.1版本

5.注意anaconda虚拟环境里的cuda和cudnn不是完整的，必须依赖于系统内安装的cuda和cudnn

6.在虚拟环境里使用gpu时候会优先调用虚拟环境里的cuda/cudnn