如何正确选择适合环境的CUDA版本

如何正确选择适合环境的CUDA版本

nvcc - V       #机器上的cuda版本

torch.version.cuda   #当前torch支持的最高cuda版本

nvidia-smi       #当前driver可支持的最高版本（必须大于所要安装的cuda版本）

在我们选择要安装的cuda和torch版本，都要满足 机器上的cuda版本≤当前torch支持的最高的cuda版本≤当前驱动器可支持的最高cuda版本 比如机器上的cuda版本是11.3的，torch上的最高支持版本是11.7的，当前驱动器可支持的最高的cuda版本是12.2的，那就是可以使用的cuda。