在 Linux 和 Windows 系统下查看 CUDA 和 cuDNN 版本的方法,包括使用 nvcc 命令

一直都比较头疼cuda与cudnn版本查看问题,两个系统不一样也不好查看,命令不通用

Linux

查看 CUDA 版本

方法一:

python 复制代码
nvcc --version

python 复制代码
nvcc -V

如果 nvcc 没有安装,那么用方法二。

方法二:

去安装目录下查看:

python 复制代码
cat /usr/local/cuda/version.txt

查看 cuDNN 版本

python 复制代码
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

如果没有,那么可能没有安装 cuDNN。

如果是为了使用 PyTorch/TensorFlow,在 Linux 服务器上推荐使用 conda 安装,使用 conda 可以很方便安装 PyTorch/TensorFlow 以及对应版本的 CUDA 和 cuDNN。

Windows

查看 CUDA 版本

在命令行中执行:

nvcc --version

或者进入 CUDA 的安装目录查看:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

查看 cuDNN 版本

进入 CUDA 的安装目录查看文件 cudnn.h :(注意修改v9.0)

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include\cudnn.h

如下所示,cuDNN 版本为 7.2.1 :

使用 PyTorch 查看 CUDA 和 cuDNN 版本

python 复制代码
import torch

print(torch.__version__)
python 复制代码
print(torch.version.cuda)
python 复制代码
print(torch.backends.cudnn.version())
相关推荐
玉石观沧海3 分钟前
高压变频器故障代码解析F67 F68
运维·经验分享·笔记·分布式·深度学习
努力学习的小廉38 分钟前
深入了解linux网络—— 自定义协议(上)
linux·服务器·网络
bcgbsh1 小时前
Linux开机启动脚本(cron 的 @reboot 特性)
linux·cron
听风吹雨yu1 小时前
RK3588从数据集到训练到部署YoloV8
linux·yolo·开源·rk3588·rknn
要做朋鱼燕2 小时前
【AES加密专题】1.AES的原理详解和加密过程
运维·网络·密码学·c·加密·aes·嵌入式工具
yunson_Liu2 小时前
jenkins更新了gitlab后出现报错
运维·gitlab·jenkins
iconball2 小时前
个人用云计算学习笔记 --19 (MariaDB服务器)
linux·运维·笔记·学习·云计算
Lynnxiaowen2 小时前
今天我们开始学习python3编程之python基础
linux·运维·python·学习
Chandler242 小时前
一图掌握 操作系统 核心要点
linux·windows·后端·系统
skywalk81633 小时前
term.everything‌ 通过终端运行任意GUI应用程序
运维