fix: install pytorch cu116 for server docker image (#882)

* chore: bump docker cuda version to 11.7.0 * fix: pytorch cu116 * fix: pytorch cu116
jina-ai · Dec 29, 2022 · 8a576c5 · 8a576c5
1 parent f3421f7
commit 8a576c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfiles/server.Dockerfile b/Dockerfiles/server.Dockerfile
@@ -24,7 +24,7 @@ RUN apt-get update \
     && ln -sf pip3 /usr/bin/pip \
     && pip install --upgrade pip \
     && pip install wheel setuptools nvidia-pyindex \
-    && pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
+    && pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
 
 COPY server ./server
 # given by builder