Skip to content

Commit d6dd33d

Browse files
author
thr3a
committed
Merge remote-tracking branch 'origin/master'
2 parents b710b3d + 060803b commit d6dd33d

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)