I have two GPUs, and GPU 0 is in using. So I want to train my model on GPU 1. However, I’ve tried to use
os.environ["CUDA_VISIBLE_DEVICES"]="1" in my script,
and
CUDA_VISIBLE_DEVICES=1

both not work.