What command are you running to train? omnt_train?
You can try setting CUDA_VISIBLE_DEVICES
export CUDA_VISIBLE_DEVICES=0,1
What command are you running to train? omnt_train?
You can try setting CUDA_VISIBLE_DEVICES
export CUDA_VISIBLE_DEVICES=0,1