force update cuda-drivers
@@ -17,6 +17,17 @@ if [[ "${FLEXCI_BRANCH:-}" == refs/pull/* ]]; then
echo "Testing Pull-Request: #${pull_req}"
fi
+# TODO(kmaehashi): Hack for CUDA 11.5 until FlexCI base image update
+if [[ "${TARGET}" == "cuda115" ]]; then
+ if [[ $(dpkg -s cuda-drivers | grep Version: | cut -d ' ' -f 2) == 470.* ]]; then
+ add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /"
+ apt-get purge -qqy "cuda-drivers*" "*nvidia*-470"
+ apt-get install -qqy "cuda-drivers"
+ modprobe -r nvidia_drm nvidia_uvm nvidia_modeset nvidia
+ nvidia-smi
+ fi
+fi
+
echo "Starting: "${TARGET}""
echo "****************************************************************************************************"
CACHE_DIR=/tmp/cupy_cache PULL_REQUEST="${pull_req}" "$(dirname ${0})/run.sh" "${TARGET}" cache_get build test 2>&1 | tee "${LOG_FILE}"
