1.查看自己要安装的cuda版本、对应的driver。对应的网址为:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
如果版本不对应,可以先根据更新显卡驱动到高版本,网址如下:https://www.nvidia.cn/Download/index.aspx?lang=cn
2.下载网址:https://developer.nvidia.com/cuda-toolkit-archive。安装cuda默认安装即可,和cudnn下载网址:https://developer.nvidia.com/rdp/cudnn-archive,二者一定那个要匹配。
3.下载torch是一定要版本对应,可以直接指定。https://download.pytorch.org/whl/cu117