Skip to content

Commit

Permalink
Docker: remove useless steps
Browse files Browse the repository at this point in the history
  • Loading branch information
Trojaner authored May 2, 2023
1 parent 1f76494 commit 880d8a7
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -13,23 +13,19 @@ RUN add-apt-repository ppa:deadsnakes/ppa && \
RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.10 3 && \
update-alternatives --config python3

RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python3
RUN python3 -m pip install -e /opt/pytorch/pytorch

RUN useradd -m -s /bin/bash appuser
USER appuser

WORKDIR /app

RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python3
RUN python3 -m pip install wheel

# Install requirements
COPY requirements.txt setup.py .
RUN python3 -m pip install --use-pep517 -U -r requirements.txt

# Upgrade to Torch 2.0
RUN python3 -m pip install --use-pep517 --no-deps -U triton==2.0.0 torch xformers==0.0.17 accelerate

# Fix missing libnvinfer7
USER root
RUN ln -s /usr/lib/x86_64-linux-gnu/libnvinfer.so /usr/lib/x86_64-linux-gnu/libnvinfer.so.7 && \
Expand Down

0 comments on commit 880d8a7

Please sign in to comment.