mirror of
https://github.com/graphdeco-inria/gaussian-splatting
synced 2025-03-31 15:50:00 +00:00
Dev: host gpu にアクセスOK
This commit is contained in:
parent
7b7c7490c1
commit
5efb40a557
25
Dockerfile
25
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"]
|
18
compose.yaml
18
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]
|
||||
|
Loading…
Reference in New Issue
Block a user