Browse Source

Update from deprecated version of Nvidia container

main
catid 8 months ago
parent
commit
8d0ee3eac5
  1. 17
      docker/Dockerfile

17
docker/Dockerfile

@ -1,8 +1,10 @@
# get the development image from nvidia cuda 12.1
FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu20.04
FROM nvidia/cuda:12.4.1-runtime-ubuntu22.04
LABEL name="instantmesh" \
maintainer="instantmesh"
LABEL name="instantmesh" maintainer="instantmesh"
# Add a volume for downloaded models
VOLUME /workspace/models
# create workspace folder and set it as working directory
RUN mkdir -p /workspace/instantmesh
@ -12,12 +14,12 @@ WORKDIR /workspace
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update && \
apt-get install -y tzdata && \
ln -fs /usr/share/zoneinfo/Asia/Shanghai /etc/localtime && \
ln -fs /usr/share/zoneinfo/America/Chicago /etc/localtime && \
dpkg-reconfigure --frontend noninteractive tzdata
# update package lists and install git, wget, vim, libegl1-mesa-dev, and libglib2.0-0
RUN apt-get update && \
apt-get install -y git wget vim libegl1-mesa-dev libglib2.0-0 unzip
apt-get install -y build-essential git wget vim libegl1-mesa-dev libglib2.0-0 unzip
# install conda
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
@ -36,7 +38,7 @@ RUN conda create -n instantmesh python=3.10 && echo "source activate instantmesh
ENV PATH /workspace/miniconda3/envs/instantmesh/bin:$PATH
RUN conda install Ninja
RUN conda install cuda -c nvidia/label/cuda-12.1.0 -y
RUN conda install cuda -c nvidia/label/cuda-12.4.1 -y
RUN pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121
RUN pip install xformers==0.0.22.post7
@ -51,8 +53,5 @@ RUN pip install -r requirements.txt
COPY . /workspace/instantmesh
# Add a volume for downloaded models
VOLUME /workspace/models
# Run the command when the container starts
CMD ["python", "app.py"]

Loading…
Cancel
Save