Dev: host gpu にアクセスOK

This commit is contained in:
Hideo TERADA 2024-06-07 16:50:06 +09:00
parent 7b7c7490c1
commit 5efb40a557
2 changed files with 37 additions and 6 deletions

View File

@ -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"]

View File

@ -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]