根据pytorch版本下载cuda

https://developer.nvidia.com/cuda-toolkit-archive

查看已经安装cuda 版本

nvcc --version

安装pytorch gpu

在线安装

https://pytorch.org/get-started/locally/

离线安装

https://download.pytorch.org/whl/torch_stable.html

pip install D:\DownloadData\xxxx.whl