1.根据自己GPU型号,安装GPU及cuDNN
Win参考:Windows安装GPU环境-CSDN博客
Ubuntu参考:Ubuntu 安装GPU环境-CSDN博客
2.安装PyTorch GPU版本
官网:https://pytorch.org/get-started/locally/

直接按照红框命令安装即可
1.根据自己GPU型号,安装GPU及cuDNN
Win参考:Windows安装GPU环境-CSDN博客
Ubuntu参考:Ubuntu 安装GPU环境-CSDN博客
2.安装PyTorch GPU版本
官网:https://pytorch.org/get-started/locally/

直接按照红框命令安装即可