你爱我像谁?码农!

导航

记录conda的一个坑 安装了GPU的Torch还是用的CPU在跑???

conda 安装Torch时,总是自动安装的CPU版的Torch

所以即使代码中有

torch.device("cuda" if torch.cuda.is_available() else "cpu")

运行时还是使用的CPU去跑,,,,,,,,然后经常出现的后果就是CPU100%+内存溢出!裂开

解决办法:

自己去清华的镜像网站下载GPU版的Torch,选择合适的版本下载安装

地址:

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

这些坑,记录一个少一个,要加油啊!

posted on 2021-03-25 20:34  你爱我像谁?码农!  阅读(770)  评论(0编辑  收藏  举报