收起
需要把CUDA安装好(具体的GPU对应不同版本,不同系统也会不一样),下载cuda安装包然后安装,接下来安装cudnn,最后,安装pytorch的GPU版本即可。验证是否安装成功请用torch.cuda.is_availble()打印查看。具体的教程网上贼多。
报告相同问题?