3条回答 默认 最新
- CSDN专家-li.kun 2021-04-07 15:58关注
需要把CUDA安装好(具体的GPU对应不同版本,不同系统也会不一样),下载cuda安装包然后安装,接下来安装cudnn,最后,安装pytorch的GPU版本即可。验证是否安装成功请用torch.cuda.is_availble()打印查看。具体的教程网上贼多。
本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 无用评论 打赏 举报
需要把CUDA安装好(具体的GPU对应不同版本,不同系统也会不一样),下载cuda安装包然后安装,接下来安装cudnn,最后,安装pytorch的GPU版本即可。验证是否安装成功请用torch.cuda.is_availble()打印查看。具体的教程网上贼多。