Ubuntu系统下Anaconda+pytorch的安装

参考

Anaconda的安装
CUDA的安装

准备

系统为：Ubuntu 20.04.2，在安装过程中选择完全安装，选择安装第三方软件。这样系统安装完成后，可以直接使用显卡驱动。
尽管驱动已经安装完成了，但深度学习需要的CUDA和cudnn模块没有安装。

CUDA安装

根据nvidia-smi提供的CUDA版本，我们可以支持11.2及以下的安装包。CUDA和cudnn应该是支持对ptorch等框架向下兼容的，但我们后续选择相关安装包时尽量和版本贴合。本次我安装的CUDA版本为10.1，cudnn是7.6.5，pytorch是py38下的支持cuda101和cudnn7.6.3的安装包，目前使用没有问题。

在https://developer.nvidia.com/cuda-toolkit-archive中选择对应的CUDA版本，使用对应18.10版本的安装包即可。