[build] Use conda cmake in two CI builds (pytorch#8864)

* use conda cmake in pytorch-linux-xenial-cuda8-cudnn6-py2 and pytorch-linux-xenial-cuda9-cudnn6-py3

* update test_expect

* add exit 1

* check cmake 3.5

* bump expect driver version

* add back space

Author: ssnl

.jenkins/pytorch/common.sh

@@ -118,3 +118,23 @@ if [[ "$BUILD_ENVIRONMENT" == *pytorch-linux-xenial-cuda9-cudnn7-py3 ]] || \
 else
   BUILD_TEST_LIBTORCH=0
 fi
+
+# Use conda cmake in some CI build. Conda cmake will be newer than our supported
+# min version 3.5, so we only do it in two builds that we know should use conda.
+if [[ "$BUILD_ENVIRONMENT" == *pytorch-linux-xenial-cuda* ]]; then
+  if [[ "$BUILD_ENVIRONMENT" == *cuda8-cudnn6-py2* ]] || \
+     [[ "$BUILD_ENVIRONMENT" == *cuda9-cudnn7-py3* ]]; then
+    if ! which conda; then
+      echo "Expected ${BUILD_ENVIRONMENT} to use conda, but 'which conda' returns empty"
+      exit 1
+    else
+      conda install -q -y cmake
+    fi
+  else
+    if ! cmake --version | grep 'cmake version 3\.5'; then
+      echo "Expected ${BUILD_ENVIRONMENT} to have cmake version 3.5.* (min support version), but 'cmake --version' returns:"
+      cmake --version
+      exit 1
+    fi
+  fi
+fi

test/expect/TestCollectEnv.test_pytorch_linux_xenial_cuda9_cudnn7_py3.expect

@@ -13,7 +13,7 @@ GPU models and configuration:
 GPU 0: Tesla M60
 GPU 1: Tesla M60
 
-Nvidia driver version: 390.X
+Nvidia driver version: 396.X
 cuDNN version: Probably one of the following:
 /usr/lib/x86_64-linux-gnu/libcudnn.so.7.1.4
 /usr/lib/x86_64-linux-gnu/libcudnn_static_v7.a

test/test_utils.py

@@ -673,7 +673,6 @@ def test_smoke(self):
         self.assertTrue(info_output.count('\n') >= 17)
 
     @unittest.skipIf('BUILD_ENVIRONMENT' not in os.environ.keys(), 'CI-only test')
-    @unittest.skip('temporarily skip to change cmake version in CI')
     def test_expect(self):
         info_output = get_pretty_env_info()