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