deps: fetch Torch wheels for same CUDA/CPU version

JBBalling · Feb 4, 2022 · 0a4ebf6 · 0a4ebf6
1 parent 249de05
commit 0a4ebf6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/Makefile b/Makefile
@@ -39,7 +39,8 @@ deps:
 	if test "$$CUDA_VERSION" = CPU; then CUDA=cpu; \
 	else IFS=. CUDA=($$CUDA_VERSION) && CUDA=cu$${CUDA[0]}$${CUDA[1]}; \
 	fi && $(PIP) install -r requirements.txt \
-	-f "https://dl.fbaipublicfiles.com/detectron2/wheels/$$CUDA/torch1.10/index.html"
+	-f "https://dl.fbaipublicfiles.com/detectron2/wheels/$$CUDA/torch1.10/index.html" \
+	-f "https://download.pytorch.org/whl/$$CUDA/torch_stable.html"
 
 # Install Python package via pip
 install: deps

diff --git a/requirements.txt b/requirements.txt
@@ -1,9 +1,9 @@
-ocrd>=2.18
+ocrd>=2.30
 click>=7.0
 scipy
 numpy>=1.17.0
 pillow>=7.1.2
 scikit-image>=0.17.2
-torch~=1.10
-torchvision
-detectron2
+torch>=1.10.2
+torchvision>=0.11.3
+detectron2>=0.6