记录conda的一个坑 安装了GPU的Torch还是用的CPU在跑???
摘要:
conda 安装Torch时,总是自动安装的CPU版的Torch 所以即使代码中有 torch.device("cuda" if torch.cuda.is_available() else "cpu") 运行时还是使用的CPU去跑,,,,,,,,然后经常出现的后果就是CPU100%+内存溢出!裂开 阅读全文
posted @ 2021-03-25 20:34 你爱我像谁?码农! 阅读(770) 评论(0) 推荐(0) 编辑