cuda가 원인인거 같은데 이미지 크기를 어떻게 줄일까요..
처음 도커 이미지를 생성하는거라 수정이 필요한 사항 조언 부탁드립니다!
# base image
FROM ubuntu:18.04
# install package
RUN apt-get update && apt-get install -y \
gcc \
g++ \
wget \
git \
cmake protobuf-compiler \
software-properties-common \
nvidia-driver-470
# dpkg & Check the installed nvidia driver package
RUN dpkg -l | grep -i nvidia
# install cuda
RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
RUN sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
RUN sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
RUN sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /"
RUN sudo apt-get update
RUN sudo apt-get -y install cuda
# Authorize SSH Host
RUN mkdir /root/.ssh/
ADD ./.ssh/id_rsa /root/.ssh/id_rsa
RUN chmod 600 /root/.ssh/id_rsa
# Add the keys and set permissions
RUN touch /root/.ssh/known_hosts
RUN ssh-keyscan gitlab.com >> /root/.ssh/known_hosts
# private repo clone
RUN git clone ***@******.com:link.git
# build
WORKDIR /link
RUN make
https://hub.docker.com/r/nvidia/cuda/tags
위 링크처럼 cuda 11.3 ubuntu 18.04 를 사용하고 싶은데 베이스 이미지를
FROM nvidia/cuda:11.3.0-base-ubuntu18.04 를 사용해도 본문의 명령어들을 정상 실행 할수 있을까요?
또, 해당 베이스 이미지를 사용하면 아래 cuda 설치 명령어들을 지워도 되는건가요?
RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
RUN sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
RUN sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
RUN sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /"
RUN sudo apt-get update
RUN sudo apt-get -y install cuda
&&로 명령들을 묶으면 용량을 줄일 수 있습니다. 대신 도커 캐시의 재사용률이 떨어지겠죠.
마지막에 apt와 dpkg 캐시 지우는건 거의 필수고
git도 마지막 커밋만 받도록 (maxdepth) 하면 git history를 받지 않습니다.