记录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/
这些坑,记录一个少一个,要加油啊!