diff --git a/docker/install/ubuntu_install_oneflow.sh b/docker/install/ubuntu_install_oneflow.sh
index 154fc225abff..3eb6b7d89bf4 100755
--- a/docker/install/ubuntu_install_oneflow.sh
+++ b/docker/install/ubuntu_install_oneflow.sh
@@ -20,4 +20,6 @@ set -e
 set -u
 set -o pipefail
 
-python3 -m pip install -f https://release.oneflow.info oneflow==0.6.0+cpu
+pip3 install flowvision==0.1.0
+
+python3 -m pip install -f https://release.oneflow.info oneflow==0.7.0+cpu
diff --git a/tests/python/frontend/oneflow/test_forward.py b/tests/python/frontend/oneflow/test_forward.py
index d144cdad2bc5..c42644cdfbbc 100644
--- a/tests/python/frontend/oneflow/test_forward.py
+++ b/tests/python/frontend/oneflow/test_forward.py
@@ -710,14 +710,14 @@ def forward(self, x1, x2, x3):
         verify_concat(model, device=device)
 
 
-if __name__ == "__main__":
-    test_conv2d()
-    test_pool2d()
-    test_normalization()
-    test_upsample()
-    test_convtran()
-    test_activation()
-    test_math()
-    test_slice()
-    test_concat()
-    rmdir("log")
+# if __name__ == "__main__":
+    # test_conv2d()
+    # test_pool2d()
+    # test_normalization()
+    # test_upsample()
+    # test_convtran()
+    # test_activation()
+    # test_math()
+    # test_slice()
+    # test_concat()
+    # rmdir("log")