GPU
https://pasongsong.tistory.com/429
도커 GPU 설정하기 Nvidia Toolkit
docker 19.*.* 부터 nvidia-docker가 아닌 nvidia-container-runtime를 설치하면 GPU인식 및 활용이 가능하다. https://docs.docker.com/config/containers/resource_constraints/#gpu docker-24.0.7 nvidia-container-runtime 설치 $ sudo apt-get ins
pasongsong.tistory.com
그 전에 도커 gpu먼저 설정해야한다.
torch image
Docker Hub Container Image Library | App Containerization
Build and Ship any Application Anywhere Docker Hub is the world's easiest way to create, manage, and deliver your team's container applications. Create your account Signing up for Docker is fast and free. Continue with GoogleContinue with GitHubContinue wi
hub.docker.com
위의 사이트에서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 시간의 예정 이 블로그를 보시는 분은 더 빠르게 설정하시길..
'기타 > Docker' 카테고리의 다른 글
| 도커 이해하기 - 도커 기초 (0) | 2023.12.01 |
|---|---|
| 우분투 도커 관련 명령어 모음집(계속 UPDATE 예정) (0) | 2023.11.29 |
| 도커 입문 - 설치/ Image/ Container (0) | 2023.11.29 |
| 도커 GPU 설정하기 Nvidia Toolkit (0) | 2023.04.21 |
GPU
https://pasongsong.tistory.com/429
도커 GPU 설정하기 Nvidia Toolkit
docker 19.*.* 부터 nvidia-docker가 아닌 nvidia-container-runtime를 설치하면 GPU인식 및 활용이 가능하다. https://docs.docker.com/config/containers/resource_constraints/#gpu docker-24.0.7 nvidia-container-runtime 설치 $ sudo apt-get ins
pasongsong.tistory.com
그 전에 도커 gpu먼저 설정해야한다.
torch image
Docker Hub Container Image Library | App Containerization
Build and Ship any Application Anywhere Docker Hub is the world's easiest way to create, manage, and deliver your team's container applications. Create your account Signing up for Docker is fast and free. Continue with GoogleContinue with GitHubContinue wi
hub.docker.com
위의 사이트에서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 시간의 예정 이 블로그를 보시는 분은 더 빠르게 설정하시길..
'기타 > Docker' 카테고리의 다른 글
| 도커 이해하기 - 도커 기초 (0) | 2023.12.01 |
|---|---|
| 우분투 도커 관련 명령어 모음집(계속 UPDATE 예정) (0) | 2023.11.29 |
| 도커 입문 - 설치/ Image/ Container (0) | 2023.11.29 |
| 도커 GPU 설정하기 Nvidia Toolkit (0) | 2023.04.21 |
