Spaces:
Running
on
T4
Running
on
T4
File size: 2,144 Bytes
e289681 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
ENV DEBIAN_FRONTEND=noninteractive
ENV FORCE_CUDA="1"
ENV CUDA_HOME="/usr/local/cuda-12.4"
ARG TORCH_CUDA_ARCH_LIST="7.5+PTX"
ENV LD_LIBRARY_PATH="${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}"
ENV PATH="/usr/local/cuda/bin:/usr/local/cuda-12/bin:${CUDA_HOME}/bin:${PATH}"
ENV HF_HOME="/data/.huggingface"
###########################
# Prepare the environment #
###########################
WORKDIR /code
RUN apt-get update && apt-get install -y \
wget \
bzip2 \
ca-certificates \
curl \
libglib2.0-0 \
libx11-6 \
git \
vim \
cmake \
make \
g++-9 \
libgl-dev \
freeglut3 \
freeglut3-dev \
libosmesa6-dev \
libglu1-mesa-dev \
libglu1-mesa \
xserver-xorg-video-dummy \
&& rm -rf /var/lib/apt/lists/*
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py310_24.5.0-0-Linux-x86_64.sh -O /tmp/miniconda.sh \
&& bash /tmp/miniconda.sh -b -p /opt/conda \
&& rm /tmp/miniconda.sh
ENV PATH="/opt/conda/bin:${PATH}"
COPY ./environment.yml /code/environment.yml
COPY ./pointnet2_ops_lib /code/pointnet2_ops_lib
RUN conda env create -f environment.yml
COPY ./requirements.txt /code/requirements.txt
SHELL ["/bin/bash", "-c"]
RUN source activate HoLa-Brep && \
export CUDA_HOME=/usr/local/cuda-12.4 && \
export FORCE_CUDA=1 && \
python -m pip install --no-cache-dir torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/cu124
RUN conda list -n HoLa-Brep | grep pytorch
RUN source activate HoLa-Brep && \
export CUDA_HOME=/usr/local/cuda-12.4 && \
export FORCE_CUDA=1 && \
python -m pip install --no-cache-dir -r /code/requirements.txt
###########
# Run app #
###########
RUN useradd -m -u 1000 user
USER user
ENV HOME="/home/user"
ENV PYTHONPATH="/home/user/HoLa-Brep:/data:${PYTHONPATH}"
EXPOSE 7860
ENV GRADIO_SERVER_NAME="0.0.0.0"
WORKDIR ${HOME}/HoLa-Brep
COPY --chown=user . ${HOME}/HoLa-Brep
CMD ["/bin/bash", "-c", "source activate HoLa-Brep && python app.py"]
|