We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ccd8a0f commit 060803bCopy full SHA for 060803b
cuda12.4-torch/Dockerfile
@@ -19,6 +19,6 @@ RUN apt-get update \
19
&& ln -nfs /usr/bin/python${PYTHON_VERSION} /usr/bin/python3 \
20
&& rm -rf /var/lib/apt/lists/* \
21
&& curl -sSL https://bootstrap.pypa.io/get-pip.py | python -
22
-RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
+RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cu124
23
24
WORKDIR /app
0 commit comments