在 Linux 和 Windows 系统下检察 CUDA 和 cuDNN 版本的方法,包括使用 nvcc ...

鼠扑  金牌会员 | 2024-6-13 21:49:59 | 显示全部楼层 | 阅读模式
打印 上一主题 下一主题

主题 577|帖子 577|积分 1731

不停都比较头疼cuda与cudnn版本检察问题,两个系统不一样也不好检察,下令不通用
Linux
检察 CUDA 版本
方法一:
  1. nvcc --version
复制代码

  1. nvcc -V
复制代码
假如 nvcc 没有安装,那么用方法二。
方法二:
去安装目次下检察:
  1. cat /usr/local/cuda/version.txt
复制代码
检察 cuDNN 版本
  1. 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 版本
  1. import torch
  2. print(torch.__version__)
复制代码
  1. print(torch.version.cuda)
复制代码
  1. print(torch.backends.cudnn.version())
复制代码
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

鼠扑

金牌会员
这个人很懒什么都没写!

标签云

快速回复 返回顶部 返回列表