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())

總結

以上是生活随笔為你收集整理的Linux查看CUDA和cuDNN版本的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。