CUDA error: no kernel image is available for execution on the device

Hi there, I’m trying to get my new RTX 3090 to work nicely with CUDA but this seems harder than expected. I think I have all the drivers set up and the correct CUDA version:

±----------------------------------------------------------------------------+
| NVIDIA-SMI 470.57.02 Driver Version: 470.57.02 CUDA Version: 11.4 |
|-------------------------------±---------------------±---------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 NVIDIA GeForce … On | 00000000:21:00.0 On | N/A |
| 0% 37C P8 22W / 350W | 957MiB / 24265MiB | 0% Default |
| | | N/A |
±------------------------------±---------------------±---------------------+

±----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
| 0 N/A N/A 1525 G /usr/lib/xorg/Xorg 53MiB |
| 0 N/A N/A 2212 G /usr/lib/xorg/Xorg 208MiB |
| 0 N/A N/A 2342 G /usr/bin/gnome-shell 94MiB |
| 0 N/A N/A 3640 G /usr/lib/firefox/firefox 570MiB |
| 0 N/A N/A 3797 G /usr/lib/firefox/firefox 4MiB |
| 0 N/A N/A 4241 G /usr/lib/firefox/firefox 4MiB |
| 0 N/A N/A 4275 G /usr/lib/firefox/firefox 4MiB |
±----------------------------------------------------------------------------+

I’m just running spacy.require_gpu()

And I get the ‘no kernel image’ error.

Any clues what I’m doing wrong?