diff --git a/Dockerfile b/Dockerfile index 55801cc..5291eda 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,3 +1,26 @@ -FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 +#FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 +FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04 +#FROM nvidia/cuda:12.5.0-devel-ubuntu22.04 + +RUN apt update -y +RUN apt install -y wget + +RUN apt install -y build-essential +RUN apt install -y python3 python3-pip python-is-python3 + +#RUN mkdir -p /miniconda3 +RUN mkdir -p /anaconda3 +# RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /miniconda3/miniconda.sh +RUN wget https://repo.anaconda.com/archive/Anaconda3-2024.02-1-Linux-x86_64.sh -O /anaconda3/anaconda.sh +# RUN bash /miniconda3/miniconda.sh -b -u -p /miniconda3 +RUN bash /anaconda3/anaconda.sh -b -u -p /anaconda3 +# RUN rm -f /miniconda3/miniconda.sh + +# RUN /miniconda3/bin/conda init bash +RUN /anaconda3/bin/conda init bash + +COPY environment.yml /root/environment.yml +#RUN /miniconda3/bin/conda env create -f /root/environment.yml +#RUN /anaconda3/bin/conda env create -f /root/environment.yml CMD ["/bin/bash"] \ No newline at end of file diff --git a/compose.yaml b/compose.yaml index 306a873..8b193d9 100644 --- a/compose.yaml +++ b/compose.yaml @@ -6,14 +6,22 @@ services: dockerfile: Dockerfile container_name: gsplat-dev-container image: gsplat:1.0 - user: 1010:1010 + # user: 1010:1010 #terada volumes: - - /etc/passwd:/etc/passwd:ro - - /etc/group:/etc/group:ro - - .:/home/terada/gaussian-splatting + # - /etc/passwd:/etc/passwd:ro + # - /etc/group:/etc/group:ro + - .:/root/gaussian-splatting - ~/.ssh:/root/.ssh # ports: # - "8087:8088" - working_dir: /home/terada/gaussian-splatting + working_dir: /root/gaussian-splatting tty: true network_mode: "host" + # host gpu を使えるようにする設定 https://docs.docker.jp/compose/gpu-support.html + deploy: + resources: + reservations: + devices: + - driver: nvidia + count: 1 + capabilities: [gpu]