Skip to content

Commit

Permalink
always update Docker torch version
Browse files Browse the repository at this point in the history
  • Loading branch information
thelinuxkid committed Dec 7, 2023
1 parent 58b3c0c commit c4ec563
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 cargo build --release
# Adapted from: https://github.com/pytorch/pytorch/blob/master/Dockerfile
FROM debian:bullseye-slim as pytorch-install

ARG PYTORCH_VERSION=2.0.0
ARG PYTORCH_VERSION=2.1.1
ARG PYTHON_VERSION=3.9
ARG CUDA_VERSION=11.8
ARG MAMBA_VERSION=23.1.0-1
Expand Down Expand Up @@ -74,7 +74,7 @@ RUN case ${TARGETPLATFORM} in \
"linux/arm64") exit 1 ;; \
*) /opt/conda/bin/conda update -y conda && \
/opt/conda/bin/conda install -y "python=3.9" && \
/opt/conda/bin/pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118 ;; \
/opt/conda/bin/pip install torch==2.1.1+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118 ;; \
esac && \
/opt/conda/bin/conda clean -ya

Expand Down

0 comments on commit c4ec563

Please sign in to comment.