本文已参与 「新人创作礼」 活动，一起开启掘金创作之路。
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
具体路径可以去看环境变量。
