upgrade to 2.3.0

sophgo · May 31, 2024 · 9486f03 · 9486f03
1 parent ad96ed7
commit 9486f03
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -1,7 +1,7 @@
-FROM bernardxu6034/sgmq_base:v1
+FROM sophgo/tpuc_dev:v3.3-cuda
 
 WORKDIR /workspace
 
 RUN echo "Setting up my images"
 
-EXPOSE 8080
+EXPOSE 8080
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
-torch==2.0.1
-torchvision==0.15.2
+torch==2.3.0
+torchvision==0.18.0
 onnx
 prettytable
 easydict