在 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())
相关推荐
linux修理工21 分钟前
ipmitool 使用简介(ipmitool sel list & ipmitool sensor list)
运维·服务器
寻月隐君23 分钟前
保姆级教程:Zsh + Oh My Zsh 终极配置,让你的 Ubuntu 终端效率倍增
linux·后端·命令行
XM-545825 分钟前
2025微信小程序wxapkg解包全攻略
linux·运维·小程序
朗晴1 小时前
文本编辑器VIM的使用方法!
linux·运维·服务器
小林up1 小时前
github push:ssh: connect to host github.com port 22
运维·ssh·github
梁bk4 小时前
[Nginx]反向代理和负载均衡
运维·nginx·负载均衡
超龄超能程序猿5 小时前
dnSpy 使用教程
windows·microsoft
2401_826097629 小时前
JavaEE-Linux环境部署
java·linux·java-ee
(:满天星:)10 小时前
第31篇:块设备与字符设备管理深度解析(基于OpenEuler 24.03)
linux·运维·服务器·网络·centos
小陶来咯10 小时前
【仿muduo库实现并发服务器】Acceptor模块
运维·服务器