Linux 查看cuda/cudnn/python/pytorch/torchvision版本

查看 CUDA 版本

方法一:

nvcc --versionnvcc -V

方法二:

去安装目录下查看:
cat /usr/local/cuda/version.txt

查看 cuDNN 版本

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

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

如果是为了使用 PyTorch/TensorFlow,在 Linux 服务器上推荐使用 conda 安装,使用 conda 可以很方便安装 PyTorch/TensorFlow 以及对应版本的 CUDA 和 cuDNN。

查看pytorch版本,使用 pytorch 查看 CUDA、cuDNN 版本

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

查看python、torchvision版本

python
import platform
print(platform.python_version()) 
import torchvision
print(torchvision.__version__)

参考链接:https://blog.csdn.net/weixin_39916758/article/details/111512561

你可能感兴趣的:(linux)