도커 인공지능 환경 구성 - NVIDIA/torch

 

GPU

https://pasongsong.tistory.com/429

도커 GPU 설정하기 Nvidia Toolkit

그 전에 도커 gpu먼저 설정해야한다.

torch image

https://hub.docker.com/

Docker Hub Container Image Library | App Containerization

위의 사이트에서torch를 찾는다.

 

자신의 cuda 버전에 맞게 pull하자!

---

다운 받았다면 docker images에 받아진 것을 볼 수 있다.

---

$ docker run --gpus all --name torch2.1.1v2 pytorch/pytorch

$ docker run --gpus all -it --name torch2.1.1v2 pytorch/pytorch  /bin/bash

--gpus를 설정하여 gpu를 잡을 수 있게 한다.

아래 코드는 run 실행과 동시에 exit 되어서 /bin/bash을 실행하게 해뒀다.

---

확인

Container 환경에서 nvidia-smi이 실행되는 것을 확인할 수 있음 

---

코드 상에서 GPU가 잡히는 것을 확인할 수 있다,,,1 시간의 예정 이 블로그를 보시는 분은 더 빠르게 설정하시길..