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

最新推荐文章于 2024-10-15 01:00:00 发布

大龙唉

最新推荐文章于 2024-10-15 01:00:00 发布

阅读量2.2k

点赞数

文章标签： linux 深度学习运维

本文链接：https://blog.csdn.net/weixin_44001371/article/details/134448625

版权

查看已经安装的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，则表示可用