diff --git a/backend/Dockerfile b/backend/Dockerfile
index 27ee5e64cf348e02584c64eeca5d175075ef0cb9..4ba3d3762e41ef14ae966cc835bfc9e468d52dde 100644
--- a/backend/Dockerfile
+++ b/backend/Dockerfile
@@ -60,7 +60,7 @@ RUN curl -SL https://github.com/LibrePhotos/librephotos-docker/releases/download
 RUN curl -SL https://github.com/LibrePhotos/librephotos-docker/releases/download/0.1/clip-embeddings.tar.gz | tar -zxC /data_models/
 RUN curl -SL https://download.pytorch.org/models/resnet152-b121ed2d.pth -o /root/.cache/torch/hub/checkpoints/resnet152-b121ed2d.pth
 
-RUN pip install torch==1.8.0+cpu torchvision==0.9.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
+RUN pip install torch==1.8.0 torchvision==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
 
 RUN pip3 install pyvips