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后,再执行上面的命令

在这里插入图片描述
结果:
pytorch安装gpu版本(torch.cuda.is_avaliable()==False解决方案)_第1张图片

你可能感兴趣的:(pytorch,深度学习,python)