1.确定pytorch和cuda的版本 pytorch官网:https://pytorch.org/ 这里选择的是cuda 11.1,torch 1.8.1。(在下载之前要先确定电脑显卡是否支持,向下是兼容的,可以在显卡控制面板的设备信息中找到) 2.下载cuda 11.1。 NVIDIA官网:https://developer.nvidia.com/zh-cn 可以去官网找,推荐直接百度cuda 这是最新版本的cuda,一般不推荐使用。在这里使用的是旧版本的11.1 3.安装和配置cud