diff --git a/docker/install/ubuntu_install_onnx.sh b/docker/install/ubuntu_install_onnx.sh
index f3e8d8e8f540..54210b83f4d6 100755
--- a/docker/install/ubuntu_install_onnx.sh
+++ b/docker/install/ubuntu_install_onnx.sh
@@ -27,5 +27,4 @@ pip3 install onnx==1.5.0
 # not expose that in the wheel!!!
 pip3 install future
 
-pip3 install https://download.pytorch.org/whl/cu80/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl
-pip3 install torchvision
+pip3 install torch==1.2.0 torchvision==0.4.0