在 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())
相关推荐
asdfg12589637 小时前
为什么要在出口路由器router配置NAT与默认路由
运维·网络·计算机网络
代码AC不AC7 小时前
【Linux】vim工具篇
linux·vim·工具详解
码农hbk7 小时前
Linux signal 图文详解(三)信号处理
linux·信号处理
bug攻城狮7 小时前
Skopeo 工具介绍与 CentOS 7 安装指南
linux·运维·centos
宇宙第一小趴菜7 小时前
08 修改自己的Centos的软件源
linux·运维·centos
bug攻城狮7 小时前
彻底禁用 CentOS 7.9 中 vi/vim 的滴滴声
linux·运维·服务器·centos·vim
苹果醋38 小时前
数据库索引设计:在 MongoDB 中创建高效索引的策略
java·运维·spring boot·mysql·nginx
XMYX-08 小时前
Linux 物理机如何区分 SSD 与 HDD ——以 DELL PERC H730 Mini 为例
linux·运维
轻松Ai享生活8 小时前
5 天学习 Linux Kernel 主要原理 | Day 3:内存管理与Swap机制
linux
轻松Ai享生活8 小时前
QAT 与 GPU 在SHA-1 运算中的优劣
linux