•
이미지 ubuntu-cuda
◦
CUDA, CUDNN 이 설치된 ubuntu 를 당겨오는 도커파일
•
이미지 ubuntu-ssh
◦
ubuntu-cuda 에 ssh 접속이 가능하도록 설정함
•
이미지 pytorch
◦
ubuntu-ssh 에 pytorch 를 설치함
•
이미지 develop
◦
볼륨을 할당해서 컨테이너가 종료되어도 개발내용이 보존되는 이미지를 생성함
•
이들을 모두 묶어 하나의 파일로 관리
•
환경
◦
develop 컨테이너의 환경은 다음을 상속받는다. develop 컨테이너가 켜져 있으면 원격지에서도 자유롭게 접속할 수 있다.
▪
ubuntu 20.04
▪
cudnn 8
▪
cuda 11.1.1
▪
pytorch 1.10.1
▪
ssh
•
실행
좌: GPU 서버 호스트, 우: GPU 서버에 존재하는 컨테이너
◦
GPU 서버 → GPU 서버의 도커 컨테이너
docker-compose up
ssh -p 11122 root@localhost
Bash
복사
◦
노트북 → VPN → GPU 서버의 도커 컨테이너
docker-compose up
ssh -p 11122 root@192.168.114.40
Bash
복사
•
TODO
컨테이너에서 nvidia-smi 확인
컨테이너의 pytorch 에서 CUDA 식별 확인
import torch
torch.cuda.is_available()
torch.cuda.device_count()
torch.cuda.current_device()
torch.cuda.device(0)
torch.cuda.get_device_name(0)
Bash
복사
컨테이너에 git 설치