训练时选择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)  编辑  收藏  举报