图像训练显卡相关备注

Cuda：NVIDIA开发的GPU计算平台，通过应用程序安装，只能用于NVIDIA自家的显卡，Cuda与显卡驱动版本有对应关系

Cudnn：GPU加速计算库文件，需要将其复制到Cuda的安装目录，能够加速GPU的训练过程，Cudnn与Cuda版本有对应关系

1.检查显卡驱动版本

显卡与cuda版本对应关系查看：Release Notes :: CUDA Toolkit Documentation

2.查看Pytorch最高支持cuda版本

Pytorch最新版本下载页面：Start Locally | PyTorch

Pytorch各详细版本下载链接：https://download.pytorch.org/whl/torch_stable.html

3.下载cuda、cudnn

查看cuda与cudnn对应版本，cuda和cudnn版本对应关系：cuDNN Archive | NVIDIA Developer