运行深度学习开源代码时选择环境经验

一、根据开源代码应用的Pytorch版本确定可用的CUDA版本列表1

在以下网站查看Pytorch对应CUDA列表：https://pytorch.org/
该列表只有列出的CUDA版本才支持，向下不兼容

二、根据显卡驱动确认可用的CUDA版本列表2

在终端输入

nvidia-smi
#显示如下
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 450.66       Driver Version: 450.66       CUDA Version: 11.0     |
|-------------------------------+----------------------+----------------------+

显卡支持的CUDA版本向下兼容

三、确定CUDA版本

根据一、二确定的两个可用CUDA版本列表选择一个同时满足两个列表的CUDA版本

四、确定CUDNN版本

根据CUDA版本确认CUDNN版本，对应关系如下：
https://developer.nvidia.com/rdp/cudnn-archive