import os os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" # Arrange GPU devices starting from 0 os.environ["CUDA_VISIBLE_DEVICES"]= "0,1" # 0번 과 1번 GPU를 사용, Set the GPU 2 to use 위의 코드를 실행할 코드 맨 위에 적용하여 실행하면 됨.

git clone https://github.com/NVIDIA/apex cd apex pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./ 아래의 주소로 가서 참조하면 된다. https://github.com/NVIDIA/apex GitHub - NVIDIA/apex: A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch A PyTorch Extension: Tools for easy mixed precision and distributed tr..