安装pytorch时出现以下问题:
需要安装GPU版本的pytorch才能使用GPU加速。安装前查看你本机GPU的版本,然后去官网(https://pytorch.org/)安装对应版本的pytorch
import torch torch.cuda.is_available()