查看CUDA和cudnn版本遇到的问题

目录

• Linux
  
  
  
  • 查看 CUDA 版本
  
  
  • 查看 cuDNN 版本
  
  
  
  


• Windows
  
  
  
  • 查看 CUDA 版本
  
  
  • 查看 cuDNN 版本
  
  
  
  


• 使用 PyTorch 查看 CUDA 和 cuDNN 版本


• References

Linux

查看 CUDA 版本

• 方法一：

nvcc --version


或

nvcc -V


如果 nvcc 没有安装，那么用方法二。

• 方法二：

去安装目录下查看：

cat /usr/local/cuda/version.txt


查看 cuDNN 版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2


如果没有，那么可能没有安装 cuDNN。

网上查到使用下面第一条指令就可查看安装好的cudnn版本，但是好像最近使用起来一直没反应，经过探索我发现，不知道从哪个版本开始，cudnn的版本号已经不在cudnn.h中了，而是单独有一个文件cudnn_version.h，因此在完整安装方法基础上，还需要在复制cudnn.h的同时也将cudnn_version.h一起复制到我们自己的cuda对应目录下，以本人使用的ubuntu18.04为例，应该是/usr/local/cuda/include/下

原指令

环境变量内查看 CUDA_PATH 或 path。

使用 PyTorch 查看 CUDA 和 cuDNN 版本

import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())


References

ubuntu: 查看cuda版本 -- JNingWei
Windows查看CUDA版本 -- 胡大炮的妖孽人生
LINUX下查询CUDA和CUDNN版本 -- 许文_北京北
Linux和Windows下查看环境变量方法 -- 小寒

作者：wuliytTaotao

出处：https://www.cnblogs.com/wuliytTaotao/

本作品采用知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议进行许可，欢迎转载，但未经作者同意必须保留此段声明，且在文章页面明显位置给出原文连接。

原文：https://www.cnblogs.com/wuliytTaotao/p/11453265.html