在 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())
相关推荐
匆匆那年96713 分钟前
Docker容器中安装MongoDB,导入数据
运维·docker·容器
望获linux17 分钟前
【Linux基础知识系列】第四十三篇 - 基础正则表达式与 grep/sed
linux·运维·服务器·开发语言·前端·操作系统·嵌入式软件
眠りたいです27 分钟前
Mysql常用内置函数,复合查询及内外连接
linux·数据库·c++·mysql
我的泪换不回玫瑰29 分钟前
Linux系统管理命令
linux
万米商云1 小时前
企业物资集采平台解决方案:跨地域、多仓库、百部门——大型企业如何用一套系统管好百万级物资?
大数据·运维·人工智能
jjkkzzzz1 小时前
Linux下的C/C++开发之操作Zookeeper
linux·zookeeper·c/c++
二当家的素材网1 小时前
Centos和麒麟系统如何每天晚上2点10分定时备份达梦数据库
linux·数据库·centos
挑战者6668881 小时前
CentOS 系统高效部署 Dify 全攻略
linux·运维·centos
DD_陈东东2 小时前
gdbus 常用数据结构与库函数
linux
DIY机器人工房3 小时前
代码详细注释:ARM-Linux字符设备驱动开发案例:LCD汉字输出改进建议开发板断电重启还能显示汉字,显示汉字位置自定义
linux·嵌入式·文件io·diy机器人工房