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

查看已经安装的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 torch

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

Linux(多用户下)查看cuda、cudnn版本、查看已经安装的cuda版本相关命令_第1张图片

查询cudnn版本

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

在这里插入图片描述

你可能感兴趣的:(linux,深度学习,运维)