Fix CUDA version test