在 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())
相关推荐
Awkwardx29 分钟前
Linux系统编程—线程同步与互斥
linux·服务器
snpgroupcn1 小时前
Adobe SAP S/4HANA 升级实践:企业规模化转型关键要素
运维·adobe·云计算
赖small强1 小时前
[Linux]内核队列实现详解
linux·kfifo·请求队列·工作队列(workqueue)·等待队列·kfifo_init
非凡ghost2 小时前
Adobe Lightroom安卓版(手机调色软件)绿色版
前端·windows·adobe·智能手机·软件需求
www.023 小时前
linux服务器升级显卡驱动(笔记)
linux·运维·服务器·笔记·ubuntu·服务器环境
准时准点睡觉3 小时前
window安装MYSQL5.5出错:a windows service with the name MYSQL alreadyexists....
数据库·windows·mysql
Yeats_Liao3 小时前
Go Web 编程快速入门 13 - 部署与运维:Docker容器化、Kubernetes编排与CI/CD
运维·前端·后端·golang
wdfk_prog3 小时前
[Linux]学习笔记系列 -- [kernel][time]hrtimer
linux·笔记·学习
摇滚侠3 小时前
Spring Boot3零基础教程,把 Java 程序打包为 Linux 可执行文件,笔记91
java·linux·笔记
xuhe24 小时前
🛠️ 深度解析我的 Overleaf 私有化部署:一份稳定、高兼容性的 `docker-compose.yaml`
linux·docker·科研·overleaf