Use torchvision 0.9.1 to avoid agent hang issue

c3sr · Jun 14, 2022 · 969502a · 969502a
1 parent 527463e
commit 969502a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -2,7 +2,7 @@ FROM c3sr/go-mxnet:amd64-cpu-mxnet1.8.0-latest
 
 
 
-RUN wget https://bootstrap.pypa.io/pip/3.6/get-pip.py && python3 get-pip.py && pip3 install opencv-python==4.2.0.32 && pip3 install numpy && pip3 install torchvision && pip3 install scipy
+RUN wget https://bootstrap.pypa.io/pip/3.6/get-pip.py && python3 get-pip.py && pip3 install opencv-python==4.2.0.32 && pip3 install numpy && pip3 install torchvision===0.9.1 && pip3 install scipy
 
 
 # Make our go directory structure