当前位置:首页 > 编程笔记 > 正文
已解决

Linux(多用户下)查看cuda、cudnn版本、查看已经安装的cuda版本相关命令

来自网友在路上 167867提问 提问时间:2023-11-19 19:17:16阅读次数: 67

最佳答案 问答题库678位专家为你答疑解惑

查看已经安装的CUDA多个版本

linux 中cuda默认安装在/usr/local目录中:
-可以使用命令:

ls -l /usr/local | grep cuda

查看该目录下有哪些cuda版本:
如果输出:

lrwxrwxrwx  1 root root   21 Dec 17  2021 cuda -> /usr/local/cuda-11.3/
drwxr-xr-x 17 root root 4096 Dec 17  2021 cuda-11.3
drwxr-xr-x  3 root root 4096 Dec 17  2021 down_cuda_files

则表示当前机器上安装了11.3 uda 版本,/usr/local/cuda是一个软链接,链接到了/usr/local/cuda-11.3目录,表示当前使用的是cuda-11.3版本。

具体的版本可以通过: nvcc -V # (V大写) 查看。

查看当前环境下使用的CUDA版本

import torchprint(torch.__version__)  # 查看torch当前版本号
print(torch.version.cuda)  # 编译当前版本的torch使用的cuda版本号
print(torch.cuda.is_available())  # 查看当前cuda是否可用于当前版本的Torch,如果输出True,则表示可用

在这里插入图片描述

查询cudnn版本

print(torch.backends.cudnn.version())

在这里插入图片描述

查看全文

99%的人还看了

猜你感兴趣

版权申明

本文"Linux(多用户下)查看cuda、cudnn版本、查看已经安装的cuda版本相关命令":http://eshow365.cn/6-39551-0.html 内容来自互联网,请自行判断内容的正确性。如有侵权请联系我们,立即删除!