/Docker_Tensorflow_Repository

✔Tensorflow Develop Docker

Primary LanguageShell

문서정보 : 2023.09.27.~30. 작성, 작성자 @SAgiKPJH


바로 사용하기

docker run -it --gpus all --name tensorflow_gpu_vscode_gitrepo -p 18081:8080 -d juhyung1021/tensorflow_gpu_vscode_gitrepo_image:23.08

# or

nvidia-docker run -it --name nvcr.io/nvidia/j7j/dcgm-exporter:3.2.5-3.1.7-ubuntu20.04 -p 18081:8080 -d juhyung1021/tensorflow_gpu_vscode_gitrepo_image:23.08

Docker_Tensorflow_Repository

  • 아래 조건을 만족해야 합니다.
    1. Tensorflow gpu를 활용하여 docker로 실행합니다.
    2. 이미지를 만듭니다. git이 기본적으로 설치되어 있습니다.
    3. 그 이미지를 실행할 때 마다 특정 git 주소에서 그 컨테이너 내부 저장소에 최신 버전을 다운받습니다.
    4. 이미지를 실행할 때마다, 최신 버전의 git 주소 파일을 다운받기 때문에, 언제 어디서든 최신 환경에서 작업할 수 있습니다.
    5. 그 컨테이너 내부 repository 저장소에 vscode로 접근 가능합니다.
    6. 최종적으로 작업이 완료되면 다시 git 업로드 할 수 있습니다.
  • 이는 저장공간을 Git에 두지만, 별도 공간은 계속 바뀔 수 있는 구조입니다.

제작자

@SAgiKPJH





Docker File 제작 및 실행

  • dockerfile 내용 
      # 기반이 될 이미지 선택 (이름 획득 : https://catalog.ngc.nvidia.com/orgs/nvidia/containers/tensorflow)
  FROM nvcr.io/nvidia/tensorflow:23.08-tf2-py3

  # 필요한 패키지 설치, cache 비우기
  RUN apt-get update && \
      apt-get install -y curl sudo && \
      apt-get clean && \
      rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*

  ENV WORKDIR="/app/git_repository"
  RUN mkdir -p ${WORKDIR}

  ### VSCode
  # 새로운 사용자 생성 및 비밀번호 설정
  ENV USER="user" \
      PASSWORD="password"
  RUN useradd -m ${USER} && echo "${USER}:${PASSWORD}" | chpasswd && adduser ${USER} sudo && \
      sed -i "/^root/ c\root:!:18291:0:99999:7:::" /etc/shadow

  # code-server 설치 및 세팅
  ENV WORKINGDIR="/home/${USER}/vscode"
  RUN curl -fsSL https://code-server.dev/install.sh | sh && \
      mkdir ${WORKINGDIR} && \
      su ${USER} -c "code-server --install-extension ms-python.python \
                              --install-extension ms-azuretools.vscode-docker" && \
      rm -rf ${WORKINGDIR}/.local ${WORKINGDIR}/.cache
  ###

  ### Git
  # Git 설치
  RUN apt-get install -y git

  # 작업 디렉토리 설정
  WORKDIR /app

  # Git 저장소 업데이트 스크립트 추가
  COPY update_git_repo.sh /app/update_git_repo.sh

  # 스크립트 실행 권한 부여
  RUN chmod +x /app/update_git_repo.sh
  ###

  # 작업 디렉토리 내에서 사용자 쓰기 권한 부여
  RUN chown -R ${USER}:${USER} ${WORKDIR}
  # 작업 디렉토리 내의 모든 하위 폴더에 대해 실행 권한 추가
  RUN find ${WORKDIR} -type d -exec chmod +x {} \;

  # code-server 시작
  USER ${USER}
  ENTRYPOINT /app/update_git_repo.sh & nohup code-server --bind-addr 0.0.0.0:8080 --auth password  ${WORKDIR}
  • docker vscode 추가 내용은 (https://github.com/SagiK-Repository/Docker_VSCode/blob/main/README.md)를 참고했습니다.
  • update_git_repo.sh 구성 
    #!/bin/bash

git clone https://github.com/SagiK-Repository/Docker_Tensorflow_Repository.git /app/git_repository
  • 다음과 같이 docker를 준비합니다. 
    docker build -t tensorflow_gpu_vscode_gitrepo_iamge .
docker run -it --gpus all --name tensorflow_gpu_vscode_gitrepo -p 18081:8080 -d tensorflow_gpu_vscode_gitrepo_iamge:latest
  • 다음과 같이 화면을 통해 git에서 받은 내용을 확인할 수 있습니다.
    image
    image