根据GPU型号确定可使用的CUDA版本进而确定其他软件的版本对深度学习、高性能并行计算都有重要的参考价值。鉴于互联网资料在此方面的总结较少,根据自身经验简单介绍版本之间的对应关系。
首先用户可以确定的是自己的GPU型号。根据自己的GPU型号可以确定该GPU对应的计算兼容性版本号:GPU型号与computer capability对应关系; 根据Compute Capability可以找到GPU可以使用的CUDAToolkit 版本号的最高兼容版本。如果确定了CUDA版本就可以很容易确定适用的驱动版本。对于Windows来说,安装驱动前也可以确定驱动需要对应的VisualStudio版本。或者对于深度学习用户来说可以确定要使用的CUDA版本和驱动后也可以确定对应的深度学习框架Torch或者TensorFlow等的版本。