配置:win10+NVIDIA驱动+python3.9
参考:Pytorch(一) Pytorch 的安装_开开心心-CSDN博客_pytorch
安装版本:cuda.v10.2 + cudnn-8.0 + Pytorch
一、检查GPU
若存在NVIDIA,电脑存在GPU,可下载cuda
点击NVIDIA控制面板,查看配置
点击帮助>系统信息>驱动程序版本,便能看到本机配置
二、下载和安装cuda
若有GPU,则下载安装对应版本的cuda和cudnn
注意:cuda、cudnn、pytorch和Pyth