cat: /usr/local/cuda/version.txt: No such file or directory?如何查询CUDA 和 CUDNN版本

 本人所用的系统为linux Ubuntu18.04,安装docker或其他框架经常要查询cuda和cudnn版本,总结如下:

查看cuda版本

方法1、

nvcc -V 
或者
nvcc --version

如果nvcc没有安装,用方法2

方法2、

cat /usr/local/cuda/version.txt

但是可能报错,提示如下:

cat: /usr/local/cuda/version.txt: No such file or directory

进入此目录,发现version.txt 换成了 version.json,可适用以下命令:

cat /usr/local/cuda/version.json

查看cudnn版本

方法1、

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

如果没反应,查询不到,可试下方法2

方法2、

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

如果以上方法不起作用,试试torch的api;
首先确定已经安装了torch,否则需要安装一下:pip install torch -i https://pypi.tuna.tsinghua.edu.cn/simple
进入python环境,输入以下代码:

import torch

#查看torch的版本
print(torch.__version__)

#查看cuda的版本
print(torch.version.cuda)

#查看cudnn的版本
print(torch.backends.cudnn.version())

有问题欢迎随时联系~

你可能感兴趣的:(Linux,deeplearning,深度学习,linux)