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

linux

查看cuda版本

  • 方法一
nvcc --version

nvcc -V

如果nvcc没有,看方法二。

  • 方法二
    去安装目录查看
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。

windows

查看cuda版本

在命令行中执行:

nvcc --version

或者进入 CUDA 的安装目录查看:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

查看cudnn版本

进入 CUDA 的安装目录查看文件 cudnn.h :(注意修改v9.0为自己的版本)

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include\cudnn.h

具体路径可以去看环境变量。

refer:https://www.cnblogs.com/wuliytTaotao/p/11453265.html

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐