우분투에서 도커를 이용하여 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...

우분투에서 도커로 파이토치를 설치하는 방법 입니다. 우분투에서 도커 설치 하는 방법은 아래 링크를 참고하세요. [Linux] 우분투(Ubuntu)에서 도커(Docker) 설치 하기 가상화 플랫폼인 Docker를 Ubuntu에서 설치하는 방법을 알아보겠습니다. 도커 공식 홈페이지에도 설치 방법이 자세히 나와 있습니다. https://docs.docker.com/engine/install/ubuntu/ Install Docker Engine on Ubuntu docs jkim83.tistory.com 파이토치 도커 이미지 다운로드 우선 도커 허브에서 파이토치 이미지를 확인합니다. $ docker search pytorch 도커 허브 사이트에서 확인해도 됩니다. https://hub.docker.com/sea..