update dockerfile

2023-01-20 09:30:14 +08:00
parent 00def97e9f
commit b35cffadbd
3 changed files with 6 additions and 6 deletions
--- a/docker/CPUDockerfile
+++ b/docker/CPUDockerfile
@@ -6,7 +6,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
    curl

 RUN pip install --upgrade pip && \
-    pip install torch==1.12.1 --extra-index-url https://download.pytorch.org/whl/cpu
+    pip install torch==1.13.1 --extra-index-url https://download.pytorch.org/whl/cpu

 ARG version