We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 13b0870 commit ee90fecCopy full SHA for ee90fec
1 file changed
spaces/jaguar-reid-megadescriptor-spherical/Dockerfile
@@ -23,8 +23,10 @@ RUN pip install --upgrade pip
23
24
ARG HYPERVIEW_VERSION=0.3.1
25
26
-# Install the released PyPI package plus the ML stack needed for timm-image.
27
-RUN pip install "hyperview[ml]==${HYPERVIEW_VERSION}" \
+# Install CPU-only PyTorch first (much smaller than default CUDA bundle),
+# then the released HyperView package with the ML extras.
28
+RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu \
29
+ && pip install "hyperview[ml]==${HYPERVIEW_VERSION}" \
30
&& python -c "import hyperview; print('hyperview', hyperview.__version__)"
31
32
COPY --chown=user demo.py ./demo.py
0 commit comments