내가 어떤 컨테이너안에서 GPU를 사용하려고 하면 문제가 발생해요. 왜냐하면 컨테이너는 OS와 격리된 상태이기 때문이죠. 그렇기 때문에 이를 해결해줄 방법이 필요한데 그것이 바로 nvidia-docker에요!
<2. NVIDIA DOCKER 설치하기>
간단하게 nvidia-docker에 대해서 말씀드릴게요.
Docker 컨테이너에서 실행되는 응용프로그램은 OS와 분리 되어 있기 때문에 docker 컨테이너내에서 gpu를 실행할 수가 없어요.
그래서 docker 내부에서 gpu (driver) 이용하기 위해서는 host gpu driver와 docker를 연동시켜주는 driver가 있어야 하는데, 그것이 바로 nividia-docker에요!
아래그림을 보시면 'nvidia driver'가 이미 설치되어 있어야 한다는점 명심해주세요!
(nividia driver는 보통 컴퓨터를 구매하고 우분투를 설치하면 설치할 수 있는데, 이 부분은 따로 우분투 설치하는 방법에 대한 글을 만들게 되면 그때 링크걸도록 할게요! 검색하셔도 많은 방법들이 나오니 참고하세요!)
6-2. Docker에서 pytorch 환경 구축하기 with jupyter notebook
안녕하세요~ 이번에는 리눅스 환경에서 Docker를 이용하여 pytorch를 구축하려고해요! (윈도우 환경에서는 GPU docker가 제공이 안되요 ㅜㅜ) (명령어 입력은 아래 사이트를 참고하고 복붙하시면될거
89douner.tistory.com
'IT > Docker' 카테고리의 다른 글
| [Docker] Docker run 옵션 추가하기 (0) | 2021.12.03 |
|---|---|
| [Docker] link 후 jupyter notebook, -p(port) 옵션 (0) | 2021.11.29 |
| [Docker] link 옵션 (0) | 2021.11.25 |
| [Docker] Systemctl 오류 (0) | 2021.11.25 |
| [Docker] Docker compose (0) | 2021.11.23 |