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