torch version

ploomber · Nov 29, 2023 · 0aa2ba4 · 0aa2ba4
1 parent cd7d04d
commit 0aa2ba4
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 0 deletions.
diff --git a/examples/docker/ocr-solara/Dockerfile b/examples/docker/ocr-solara/Dockerfile
@@ -2,6 +2,7 @@ FROM python:3.11
 
 COPY app.py app.py
 COPY label.png label.png
+RUN pip install torch==2.1.1 torchvision==0.16.1 --index-url https://download.pytorch.org/whl/cpu
 RUN pip install solara easyocr
 
 

diff --git a/examples/docker/ocr-solara/app.zip b/examples/docker/ocr-solara/app.zip