bump torch version (#25)

FelixKirsch · web-flow · commit 74143f250b7e · 2023-01-31T17:54:31.000+01:00
* new parent image

* fixes parent image version number

* sets correct requirements

* fixes dependency error with torchvision

* fixes start script
diff --git a/Dockerfile b/Dockerfile
@@ -1,4 +1,4 @@
-FROM kernai/refinery-parent-images:v1.7.0-torch-cpu
+FROM kernai/refinery-parent-images:v1.8.0-torch-cpu
 
 WORKDIR /program
 
diff --git a/dev.Dockerfile b/dev.Dockerfile
@@ -1,4 +1,4 @@
-FROM kernai/refinery-parent-images:v1.7.0-torch-cpu
+FROM kernai/refinery-parent-images:v1.8.0-torch-cpu
 
 WORKDIR /app
 
diff --git a/gpu-requirements.txt b/gpu-requirements.txt
@@ -210,13 +210,13 @@ tokenizers==0.13.2
     # via
     #   -r requirements/torch-cuda-requirements.txt
     #   transformers
-torch==1.12.1
+torch==1.13.1
     # via
     #   -r requirements/torch-cuda-requirements.txt
     #   embedders
     #   sentence-transformers
     #   torchvision
-torchvision==0.13.1
+torchvision==0.14.1
     # via
     #   -r requirements/gpu-requirements.in
     #   sentence-transformers
diff --git a/gpu.Dockerfile b/gpu.Dockerfile
@@ -1,4 +1,4 @@
-FROM kernai/refinery-parent-images:v1.7.0-torch-cuda
+FROM kernai/refinery-parent-images:v1.8.0-torch-cuda
 
 WORKDIR /program
 
diff --git a/requirements.txt b/requirements.txt
@@ -210,14 +210,16 @@ tokenizers==0.13.2
     # via
     #   -r requirements/torch-cpu-requirements.txt
     #   transformers
-torch==1.12.1
+torch==1.13.1
     # via
     #   -r requirements/torch-cpu-requirements.txt
     #   embedders
     #   sentence-transformers
     #   torchvision
-torchvision==0.13.1
-    # via sentence-transformers
+torchvision==0.14.1
+    # via
+    #   -r requirements/requirements.in
+    #   sentence-transformers
 tqdm==4.64.1
     # via
     #   -r requirements/torch-cpu-requirements.txt
diff --git a/requirements/gpu-requirements.in b/requirements/gpu-requirements.in
@@ -1,4 +1,4 @@
 -r torch-cuda-requirements.txt
 embedders==0.0.18
 spacy==3.4.2
-torchvision==0.13.1 # define version for torchvision to avoid dependency conflict
+torchvision==0.14.1 # define version for torchvision to avoid dependency conflict
diff --git a/requirements/requirements.in b/requirements/requirements.in
@@ -1,3 +1,4 @@
 -r torch-cpu-requirements.txt
 embedders==0.0.18
-spacy==3.4.2
+spacy==3.4.2
+torchvision==0.14.1 # define version for torchvision to avoid dependency conflict
diff --git a/requirements/torch-cpu-requirements.txt b/requirements/torch-cpu-requirements.txt
@@ -88,7 +88,7 @@ starlette==0.20.4
     # via fastapi
 tokenizers==0.13.2
     # via transformers
-torch==1.12.1
+torch==1.13.1
     # via -r requirements/torch-cpu-requirements.in
 tqdm==4.64.1
     # via
diff --git a/requirements/torch-cuda-requirements.txt b/requirements/torch-cuda-requirements.txt
@@ -88,7 +88,7 @@ starlette==0.20.4
     # via fastapi
 tokenizers==0.13.2
     # via transformers
-torch==1.12.1
+torch==1.13.1
     # via -r requirements/torch-cuda-requirements.in
 tqdm==4.64.1
     # via
diff --git a/start b/start
@@ -28,7 +28,7 @@ then
 fi
 
 INFERENCE_DIR=${PWD%/*}/dev-setup/inference/
-if [ ! -d "$_DIR" ] 
+if [ ! -d "$INFERENCE_DIR" ] 
 then
     INFERENCE_DIR=${PWD%/*/*}/dev-setup/inference/
     if [ ! -d "$INFERENCE_DIR" ] 

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-FROM kernai/refinery-parent-images:v1.7.0-torch-cpu`
	`1`	`+FROM kernai/refinery-parent-images:v1.8.0-torch-cpu`
`2`	`2`
`3`	`3`	`WORKDIR /program`
`4`	`4`