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.
2 parents b710b3d + 060803b commit d6dd33dCopy full SHA for d6dd33d
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