docker run -it --runtime=nvidia -u `id -u $USER`:`id -g $USER` -v $PWD:/app tensorflow/tensorflow:2.0.0-gpu-py3
docker run -it --runtime=nvidia -v $PWD:/app -d --name star tensorflow/tensorflow:2.0.0-gpu-py3
nvidia-docker run -it --ipc=host deep-colorization
-it: 안 붙이면 컨테이너 실행되고 bash shell이 열리지 않음
—runtime=nvidia: gpu를 사용하는 경우에는 붙여줘야 함
-v $PWD:/app: 볼륨 옵션, 현재 디렉토리를 컨테이너 안에 있는 /app 디렉토리로 가져와라
Dockerfile 만들기
FROM tensorflow/tensorflow:2.0.0-gpu-py3
RUN apt-get update -y && apt-get upgrade -y
RUN apt-get install -y vim && apt-get install -y git && apt-get install -y htop && apt-get install -y ack && apt-get install -y sudo
RUN apt-get install -y libsm6 libxext6 && apt-get install -y libxrender1 libfontconfig1
WORKDIR /root
RUN git clone <https://github.com/VundleVim/Vundle.vim.git> ~/.vim/bundle/Vundle.vim
COPY .vimrc /root/.vimrc
RUN echo ":qa" | vi .vimrc
RUN pip install opencv-python matplotlib tensorflow_datasets tensorflow-addons
RUN adduser --uid 500 --disabled-password --gecos '' eon
RUN adduser eon sudo
RUN cp /root/.vimrc /home/eon/
RUN chown eon:eon /home/eon/.vimrc
RUN echo "eon:12345" | chpasswd
RUN su eon
RUN echo ":qa" | vi /home/eon/.vimrc
WORKDIR /home/eon
WORKDIR → 도커 시작 후 처음 위치 설정
파일 만든후
docker build -t thisisiron/image1:tag1 .
docker build -t thisisiron/tensorflow:2.0.0-gpu-py3 --build-arg USER_ID=$(id -u) --build-arg GROUP_ID=$(id -g) .
docker push thisisiron/image1:tag1
push → docker hub에 올리기