2021年12月17日
训练时选择GPU
摘要： 1 在代码中通过GPU index指定device; 2 启动时指定： CUDA_VISIBLE_DEVICES=0,1,2,3 python train.py 阅读全文
posted @ 2021-12-17 11:35 weller 阅读(50) 评论(0) 推荐(0) 编辑