Linux(多用户下)查看cuda、cudnn版本、查看已经安装的cuda版本、切换不同版本cuda之间的切换等相关命令

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

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

lrwxrwxrwx  1 root root   21 5月  19  2021 cuda -> /usr/local/cuda-11.0/
drwxr-xr-x 18 root root 4096 11月 11  2020 cuda-10.1
drwxr-xr-x  4 root root 4096 5月  19  2021 cuda-11.0

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

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

1)如果使用pytorch,可以使用如下语句查询是否可用:

import torch

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

在这里插入图片描述

  • 如果nvidia-smi命令列出的CUDA版本与nvcc -V列出的版本号不一致,可能是由以下原因之一引起的:

1)安装多版本cuda后,还没有刷新环境变量,刷新即可;

2)CUDA有两种API,分别是运行时API和驱动API,即所谓的Runtime API与Driver API,nvidia-smi的结果除了有GPU驱动版本型号,还有CUDA Driver API的版本号,这里是10.0,而nvcc的结果是对应CUDA Runtime API

补充说明:在安装CUDA 时候会安装3大组件,分别是 NVIDIA 驱动、toolkit和samples。NVIDIA驱动是用来控制GPU硬件,toolkit里面包括nvcc编译器等,samples或者说SDK 里面包括很多样例程序包括查询设备、带宽测试等等。上面说的CUDADriver API是依赖于NVIDIA驱动安装的,而CUDA Runtime API 是通过CUDA toolkit安装的。

1、查看GPU使用情况:

命令: nvidia-smi

在这里插入图片描述

2、查看 CUDA 版本:

cat /usr/local/cuda/version.txt

在这里插入图片描述
cuda 版本为:10.1.243

3、查看 CUDNN 版本

cat /usr/local/**cuda-11.x**/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

cuda-11.x 实际的安装版本
在这里插入图片描述
cudnn 版本为:7.5.1

  • Windows下:

在这里插入图片描述

4、 不同版本cuda的切换(需要有足够的权限)

切换 = ①删除原始软链接 + ②建立新的软链接

①删除之前创建的软链接

rm -rf /usr/local/cuda  

②建立新的软链接, 将 cuda10.1 连接到 默认cuda上

sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda  

②再次核查 cuda版本:

cat /usr/local/cuda/version.txt

在这里插入图片描述

⑤查看 Linux下已安装的所有 cuda 版本

可以通过安装位置文件夹来识别已经装的cuda版本,如下:

查看  /usr/local/  文件夹下存在的不同cuda版本文件夹 

在这里插入图片描述

⑥ 更改 cudnn 版本

重新官网下载对应cuda匹配的cudnn版本重新安装,步骤如下,
以cudnn-10.0-Linux-7.4.1为例:

终端下,切换到cudnn文件所在目录,通过tar -xzvf cudnn-10.0-linux-x64-v7.4.1.5.tgz命令解压文件,会得到一个cuda文件夹,逐一执行下面的命令进行cudnn的安装:

# ① 复制文件夹内容
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

完成后,通过下面的命令查看安装情况:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

如果结果如下图逐行显示版本号,则安装成功。

在这里插入图片描述

  • 参考:

https://blog.csdn.net/anshiquanshu/article/details/120025826

  • 71
    点赞
  • 297
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值