그래도해야지어떡해

각 gpu 사용량 확인하기 nvidia-smi 각 gpu의 상태 확인하기(아래 사진에서는 10개(0-9번)의 gpu 사용 가능) 10개의 gpu 중 2번 gpu만 사용하기(2번 gpu 할당하기) # Set the GPU 2 to use os.environ["CUDA_VISIBLE_DEVICES"] = "2" # Train on the GPU or on the CPU, if a GPU is not available device = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu') print('Device:', device) print('Current cuda device:', torch.cuda.current_devic..