Quick fix of CUDA version to avoid NaNs.