Skip to content

Commit 19bf772

Browse files
committed
fix: use --index-url for CUDA torch reinstall in GPU Docker images; bump to 0.17.5
1 parent 433b3f0 commit 19bf772

3 files changed

Lines changed: 3 additions & 3 deletions

File tree

docker/ludwig-gpu/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ RUN if [ -n "${LUDWIG_VERSION}" ]; then \
3636
else \
3737
pip install --no-cache-dir '.[full]' --extra-index-url https://download.pytorch.org/whl/cu124; \
3838
fi
39-
RUN pip install --no-cache-dir --force-reinstall torch==2.12.0 torchvision==0.27.0 torchaudio==2.11.0 --extra-index-url https://download.pytorch.org/whl/cu124
39+
RUN pip install --no-cache-dir --force-reinstall torch==2.12.0 torchvision==0.27.0 torchaudio==2.11.0 --index-url https://download.pytorch.org/whl/cu124
4040

4141
WORKDIR /data
4242

docker/ludwig-ray-gpu/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,4 +37,4 @@ RUN if [ -n "${LUDWIG_VERSION}" ]; then \
3737
else \
3838
pip install --no-cache-dir '.[full]' --extra-index-url https://download.pytorch.org/whl/cu124; \
3939
fi
40-
RUN pip install --no-cache-dir --force-reinstall torch==2.12.0 torchvision==0.27.0 torchaudio==2.11.0 --extra-index-url https://download.pytorch.org/whl/cu124
40+
RUN pip install --no-cache-dir --force-reinstall torch==2.12.0 torchvision==0.27.0 torchaudio==2.11.0 --index-url https://download.pytorch.org/whl/cu124

ludwig/globals.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
# limitations under the License.
1515
# ==============================================================================
1616

17-
LUDWIG_VERSION = "0.17.4"
17+
LUDWIG_VERSION = "0.17.5"
1818

1919
MODEL_FILE_NAME = "model"
2020
MODEL_WEIGHTS_FILE_NAME = "model_weights" # legacy pickle format

0 commit comments

Comments
 (0)