指定gup训练

指定gpu训练命令

CUDA_VISIBLE_DEVICES=1 python train.py

posted @ 2021-10-01 12:42  赵立敏  阅读(54)  评论(0)    收藏  举报