是否有任何快速命令或脚本来检查安装的CUDA版本?
我在安装目录下找到了4.0的手册,但我不确定它是否是实际安装的版本。
是否有任何快速命令或脚本来检查安装的CUDA版本?
我在安装目录下找到了4.0的手册,但我不确定它是否是实际安装的版本。
当前回答
除了上面提到的那些,CUDA安装路径(如果在安装过程中没有更改)通常包含版本号
执行哪个nvcc应该提供路径,这将为您提供版本
PS:这是一种快速而肮脏的方式,上面的答案更优雅,并将通过相当大的努力产生正确的版本
其他回答
在我的cuda-11.60安装中,可以在/usr/local/cuda/version.json中找到信息。它包含完整的版本号(11.6.0而不是nvidia-smi显示的11.6)。
可按如下方式检索信息:
python -c 'import json; print(json.load(open("/usr/local/cuda/version.json"))["cuda"]["version"])'
如果您在linux上运行:
dpkg -l | grep cuda
[编辑的答案。感谢所有更正的人]
如果你跑步
nvidia-smi
您应该在命令输出的右上角找到安装的驱动程序支持的CUDA版本最高CUDA版本。至少我发现CUDA版本10.0的输出。,
您可以使用
nvcc -V
或者你可以使用
nvcc --version
或者您可以检查CUDA使用的位置
whereis cuda
然后做
cat location/of/cuda/you/got/from/above/command
如果您安装了PyTorch,只需在IDE中运行以下代码:
import torch
print(torch.version.cuda)