查看 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/TensorFlow,在 Linux 服务器上推荐使用 conda 安装,使用 conda 可以很方便安装 PyTorch/TensorFlow 以及对应版本的 CUDA 和 cuDNN。
查看pytorch版本,使用 pytorch 查看 CUDA、cuDNN 版本
python
import torch
print(torch.__version__)
print(torch.version.cuda