pytorch安装gpu版本(torch.cuda.is_avaliable()==False解决方案)

pytorch安装gpu版本踩坑记录(torch.cuda.is_avaliable()==False解决方案)

记录下配置pytorch环境遇到的一个小坑

采用

conda install pytorch

时默认安装的是cpuonly版本的

在这里插入图片描述
在这里插入图片描述
改成

conda install pytorch torchvision torchaudio cudatoolkit=11.3

注:如果已经安装了cpu版的pytorch应首先执行pip uninstall torch、conda uninstall pytorch后,再执行上面的命令

在这里插入图片描述
结果:
在这里插入图片描述

版权声明:本文为博主支棱起来哇原创文章,版权归属原作者,如果侵权,请联系我们删除!

原文链接:https://blog.csdn.net/s13679203258/article/details/123048316

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
心中带点小风骚的头像心中带点小风骚普通用户
上一篇 2022年2月24日
下一篇 2022年2月24日

相关推荐