우분투에서 도커를 이용하여 GPU를 지원하는 파이토치 설치하는 방법 입니다. 도커 이미지 다운로드 쿠다 버전 확인하기 쿠다 버전에 맞는 파이토치 도커 이미지를 다운로드 받기 위해 우선 Cuda 버전을 확인합니다. 터미널에서 아래 명령을 통해 쿠다 버전을 확인할 수 있습니다. $ nvcc -V 파이토치 GPU 도커 이미지 가져오기 제 쿠다 버전은 11.3이기 때문에 Cuda 11.3 버전의 파이토치 도커 이미지를 pull 땡기겠습니다. 쿠다 버전에 맞는 도커 이미지 태그는 아래 링크에서 확인해보시면 됩니다. https://hub.docker.com/r/pytorch/pytorch/tags Docker hub.docker.com $ docker pull pytorch/pytorch:1.12.1-cuda11...