RTX PRO 4000 Blackwell Xid 8 / CUDA_ERROR_LAUNCH_TIMEOUT in 2D Classification and Ab-initio

@wtempel yes:

$ nvidia-smi
Mon Aug 17 21:17:53 2026       
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 595.84                 Driver Version: 595.84         CUDA Version: 13.2     |
+-----------------------------------------+------------------------+----------------------+
| 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 RTX PRO 4000 Blac...    Off |   00000000:01:00.0  On |                  Off |
| 30%   41C    P1             29W /  145W |     524MiB /  24467MiB |      0%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+
|   1  NVIDIA RTX PRO 4000 Blac...    Off |   00000000:61:00.0 Off |                  Off |
| 30%   28C    P8              2W /  145W |      18MiB /  24467MiB |      0%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+

+-----------------------------------------------------------------------------------------+
| Processes:                                                                              |
|  GPU   GI   CI              PID   Type   Process name                        GPU Memory |
|        ID   ID                                                               Usage      |
|=========================================================================================|
|    0   N/A  N/A            2332      G   /usr/lib/xorg/Xorg                      107MiB |
|    0   N/A  N/A            2499    C+G   ...c/gnome-remote-desktop-daemon        225MiB |
|    0   N/A  N/A            2564      G   /usr/bin/gnome-shell                    120MiB |
|    0   N/A  N/A           80913      G   /usr/bin/nautilus                        14MiB |
|    1   N/A  N/A            2332      G   /usr/lib/xorg/Xorg                        4MiB |
+-----------------------------------------------------------------------------------------+