need to update CUDA driver in cuda115.multi
@@ -24,7 +24,7 @@ if [[ "${FLEXCI_BRANCH:-}" == refs/pull/* ]]; then
fi
# TODO(kmaehashi): Hack for CUDA 11.5 until FlexCI base image update
-if [[ "${TARGET}" == "cuda115" ]]; then
+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"
