一、根据开源代码应用的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