Linux查看CUDA和cuDNN版本

查看CUDA版本

方法一：

nvcc -version

或者

nvcc -V

方法二：

去安装目录下查看：

cat /usr/local/cuda/version.txt

查看cuDNN版本

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

如何没有，则可能没有安装cuDNN

使用 PyTorch 查看 CUDA 和 cuDNN 版本

import torch
print(torch.__version__)

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

版权声明：本文为jimlau原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://www.cnblogs.com/jimlau/p/13260269.html