Spaces:

gabejacobs
/

AudioCraftPlus-v2.0.1d

Paused

gabejacobs commited on Oct 11, 2023

Commit

88014f0

•

1 Parent(s): 156ea5f

upgrade torch

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -17,7 +17,7 @@ RUN --mount=type=cache,target=/var/cache/apt --mount=type=cache,target=/var/lib/
 RUN useradd -m -u 1000 ac
 RUN --mount=type=cache,target=/root/.cache python -m pip install --upgrade pip wheel
-ENV TORCH_COMMAND="pip install torch==2.0.1+cu121 torchaudio --extra-index-url https://download.pytorch.org/whl/cu121"
 RUN --mount=type=cache,target=/root/.cache python -m $TORCH_COMMAND
 RUN ln -s /usr/lib/x86_64-linux-gnu/libnvrtc.so.11.2 /usr/lib/x86_64-linux-gnu/libnvrtc.so
 USER 1000

 RUN useradd -m -u 1000 ac
 RUN --mount=type=cache,target=/root/.cache python -m pip install --upgrade pip wheel
+ENV TORCH_COMMAND="pip install torch==2.1.0+cu121 torchaudio --extra-index-url https://download.pytorch.org/whl/cu121"
 RUN --mount=type=cache,target=/root/.cache python -m $TORCH_COMMAND
 RUN ln -s /usr/lib/x86_64-linux-gnu/libnvrtc.so.11.2 /usr/lib/x86_64-linux-gnu/libnvrtc.so
 USER 1000