指定gup训练

指定gpu训练命令

CUDA_VISIBLE_DEVICES=1 python train.py

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