Install spaCy with correct cuda version on Colab with GPU

spacy_colab_gpu.py

# SETUP
cuda_version = !nvcc --version | grep -Po '^.*release \K([0-9]+\.[0-9])' | sed 's/\.//g'

# REQUIREMENTS
from pathlib import Path
reqs = f"""
spacy[cuda{cuda_version.s},transformers,lookups]==3.2.1
"""
Path("requirements.txt").write_text(reqs)
!pip install --quiet -r requirements.txt

# Check GPU is available
import spacy
assert spacy.require_gpu()