install proper version in chx tests

chainer · Aug 2, 2020 · 5a3829f · 5a3829f
1 parent 16ec91f
commit 5a3829f
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/.flexci/pytest_script.sh b/.flexci/pytest_script.sh
@@ -13,24 +13,33 @@ if [ ${CHAINERX} -gt 0 ]; then
         case ${CHAINER} in
             stable)
                 DOCKER_IMAGE=asia.gcr.io/pfn-public-ci/chainer-chem-py$PYTHON-chainerx-gpu-stable:latest
+                echo pip install chainer >> install.sh
+                echo pip install cupy-cuda101 >> install.sh
                 ;;
             latest)
                 DOCKER_IMAGE=asia.gcr.io/pfn-public-ci/chainer-chem-py$PYTHON-chainerx-gpu-latest:latest
+                echo pip install --pre chainer >> install.sh
+                echo pip install --pre cupy-cuda101 >> install.sh
                 ;;
             base)
                 DOCKER_IMAGE=asia.gcr.io/pfn-public-ci/chainer-chem-py$PYTHON-chainerx-gpu-base:latest
+                echo pip install chainer==${BASE} >> install.sh
+                echo pip install cupy-cuda101==${BASE} >> install.sh
                 ;;
         esac
     else
         case ${CHAINER} in
             stable)
                 DOCKER_IMAGE=asia.gcr.io/pfn-public-ci/chainer-chem-py$PYTHON-chainerx-cpu-stable:latest
+                echo pip install chainer >> install.sh
                 ;;
             latest)
                 DOCKER_IMAGE=asia.gcr.io/pfn-public-ci/chainer-chem-py$PYTHON-chainerx-cpu-latest:latest
+                echo pip install --pre chainer >> install.sh
                 ;;
             base)
                 DOCKER_IMAGE=asia.gcr.io/pfn-public-ci/chainer-chem-py$PYTHON-chainerx-cpu-base:latest
+                echo pip install chainer==${BASE} >> install.sh
                 ;;
         esac
     fi