PyTorch GPU版本安装

1.根据自己GPU型号，安装GPU及cuDNN

Win参考：Windows安装GPU环境-CSDN博客

Ubuntu参考：Ubuntu 安装GPU环境-CSDN博客

2.安装PyTorch GPU版本

官网：https://pytorch.org/get-started/locally/

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