Linux安装pytorch如何查看对应的cuda本版(两种cuda版本应该如何选择?)

Linux安装pytorch如何查看对应的cuda本版(两种cuda版本应该如何选择？)

两种cuda版本：
1、nvcc -V（选和该版本对应的cuda）
或者：cat /usr/local/cuda/version.txt
这条命令是查看当前实际安装的cuda版本，MMCV编译的时候是根据nvcc -V中的cuda版本来的。

2、nvidia-smi ：无用cuda（比实际安装的cuda版本大）
该命令可以查看当前NVIDIA驱动的版本，以及查看与此驱动相匹配的Cuda版本。