From 8c98a6e035c950d381aa36a5347503259c4aa3e8 Mon Sep 17 00:00:00 2001
From: Niaz <tigrexspalterlp@gmail.com>
Date: Thu, 19 Aug 2021 19:23:08 +0200
Subject: [PATCH] Update Dockerfile

---
 backend/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/backend/Dockerfile b/backend/Dockerfile
index 27ee5e6..4ba3d37 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
 
-- 
GitLab