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

原创已于 2023-10-06 13:22:09 修改 · 609 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #pytorch

于 2023-10-06 13:20:38 首次发布

博客主要围绕Linux安装pytorch时如何查看对应的cuda版本及两种cuda版本的选择展开。介绍了查看当前实际安装cuda版本的命令，MMCV编译依据此版本；还提到查看NVIDIA驱动版本及匹配Cuda版本的命令。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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版本。