Skip to content

Commit 060803b

Browse files
author
thr3a
committed
fix:
1 parent ccd8a0f commit 060803b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cuda12.4-torch/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,6 @@ RUN apt-get update \
1919
&& ln -nfs /usr/bin/python${PYTHON_VERSION} /usr/bin/python3 \
2020
&& rm -rf /var/lib/apt/lists/* \
2121
&& curl -sSL https://bootstrap.pypa.io/get-pip.py | python -
22-
RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
22+
RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cu124
2323

2424
WORKDIR /app

0 commit comments

Comments
 (0)