Skip to content

Commit

Permalink
Bump torch dependencies
Browse files Browse the repository at this point in the history
Closes #54
  • Loading branch information
Atinoda committed Jul 26, 2024
1 parent 05f2c93 commit 714f6a3
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ RUN cp -ar /src /app
# Base
FROM app_base AS app_nvidia
# Install pytorch for CUDA 12.1
RUN pip3 install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 \
RUN pip3 install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 \
--index-url https://download.pytorch.org/whl/cu121
# Install oobabooga/text-generation-webui
RUN pip3 install -r /app/requirements.txt
Expand All @@ -53,7 +53,7 @@ RUN chmod +x /scripts/build_extensions.sh && \
# Base
FROM app_base AS app_rocm
# Install pytorch for ROCM
RUN pip3 install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 \
RUN pip3 install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 \
--index-url https://download.pytorch.org/whl/rocm5.6
# Install oobabooga/text-generation-webui
RUN pip3 install -r /app/requirements_amd.txt
Expand Down

0 comments on commit 714f6a3

Please sign in to comment.