查看cuda和cudnn版本，检查能否被pytorch使用

查看cuda和cudnn版本

查看cuda版本

打开命令行，输入

nvcc -V

出现下图即为正常安装，版本为11.1

若提示命令行不存在，多半是环境变量没设好
需要在bin后加个\才行

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin\

查看cudnn版本

将这两个文件拖入命令行，最终都是PASS，就是cuda和cudnn安装成功

检查两个能否被pytorch使用

要检查GPU驱动程序和CUDA是否能够被PyTorch启用和访问，需要运行以下命令以返回CUDA驱动程序是否已启用：
运行代码：

import torch
print(torch.cuda.is_available()) # torch能否使用cuda
print(torch.backends.cudnn.version()) # torch使用的cudnn版本
print(torch.__version__) # torch的版本
print(torch.version.cuda) # cuda的版本

输出下面结果就是成功