GAN실험 - FreezeG

학습

1. git clone

2. pytorch 요구하므로 nividia에서 pytorch 이미지 pull

docker pull nvcr.io/nvidia/pytorch:21.09-py3

3. docker container

NV_GPU=1 nvidia-docker run --name bernice-FreezeG -it -v /etc/group:/etc/group:ro -v /etc/passwd:/etc/passwd:ro -v $(pwd):/workspace -v $(readlink -f disk1):/disk1 bernice-psp-pytorch /bin/bash

• --rm : 컨테이너 종료시 자동 삭제 (현재는 설정되어 있지 않은 옵션)
• --name bernice-FreezeG : container 이름 설정
• --user 1002:1002 : 유저 설정 (현재는 설정되어 있지 않은 옵션: 학습 종료 후 생성된 파일들 권한 직접 바꿔주기로)
• -it : 컨테이너 종료하지 않은 채로 터미널에 입력 넘겨줄 수 있음
• -v /etc/group:/etc/group:ro -v /etc/passwd:/etc/passwd:ro : 유저 정보를 읽기 전용으로 마운트
• $(pwd):/workspace -v $(readlink -f disk1):/disk1 : 필요한 폴더 마운트
• bernice-psp-pytorch : 실행시킬 이미지명
• /bin/bash : bash 로 실행

4. train

NV_GPU=1 nvidia-docker run --name bernice-FreezeG -it -v /etc/group:/etc/group:ro -v /etc/passwd:/etc/passwd:ro -v $(pwd):/workspace -v $(readlink -f disk1):/disk1 nvcr.io/nvidia/pytorch /bin/bash