mirror of
https://github.com/graphdeco-inria/gaussian-splatting
synced 2025-04-02 00:34:58 +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"]
|
CMD ["/bin/bash"]
|
18
compose.yaml
18
compose.yaml
@ -6,14 +6,22 @@ services:
|
|||||||
dockerfile: Dockerfile
|
dockerfile: Dockerfile
|
||||||
container_name: gsplat-dev-container
|
container_name: gsplat-dev-container
|
||||||
image: gsplat:1.0
|
image: gsplat:1.0
|
||||||
user: 1010:1010
|
# user: 1010:1010 #terada
|
||||||
volumes:
|
volumes:
|
||||||
- /etc/passwd:/etc/passwd:ro
|
# - /etc/passwd:/etc/passwd:ro
|
||||||
- /etc/group:/etc/group:ro
|
# - /etc/group:/etc/group:ro
|
||||||
- .:/home/terada/gaussian-splatting
|
- .:/root/gaussian-splatting
|
||||||
- ~/.ssh:/root/.ssh
|
- ~/.ssh:/root/.ssh
|
||||||
# ports:
|
# ports:
|
||||||
# - "8087:8088"
|
# - "8087:8088"
|
||||||
working_dir: /home/terada/gaussian-splatting
|
working_dir: /root/gaussian-splatting
|
||||||
tty: true
|
tty: true
|
||||||
network_mode: "host"
|
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