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 위의 코드를 실행할 코드 맨 위에 적용하여 실행하면 됨.

Pytorch 홈페이지(https://pytorch.org/)에 가면 원하는 사양에 맞춰서 설치 할 수 있다. Previous versions of PyTorch를 누르면 이전 버전 Pytorch 들을 확인 할 수 있다. Linux Conda Python CUDA 11.3 conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch Linux Conda Python CUDA 10.2 conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch