TA0 v3.21.08 - pycuda._driver.LogicError: cuInit failed: system not yet initialized

Thanks for the info. Usually this kind of error can be fixed after reinstalling the nvidia-driver again.

sudo apt purge nvidia-driver-535
sudo apt autoremove
sudo apt autoclean
sudo apt install nvidia-driver-535