需要把CUDA安装好(具体的GPU对应不同版本,不同系统也会不一样),下载cuda安装包然后安装,接下来安装cudnn,最后,安装pytorch的GPU版本即可。验证是否安装成功请用torch.cuda.is_availble()打印查看。具体的教程网上贼多。
要修改一下配置文件,默认情况下单CPU的电脑不支持。
您好,我是有问必答小助手,你的问题已经有小伙伴为您解答了问题,您看下是否解决了您的问题,可以追评进行沟通哦~
如果有您比较满意的答案 / 帮您提供解决思路的答案,可以点击【采纳】按钮,给回答的小伙伴一些鼓励哦~~
ps:问答VIP仅需29元,即可享受5次/月 有问必答服务,了解详情>>>https://vip.csdn.net/askvip?utm_source=1146287632