Make sure nvidia-smi detects all devices and doesn’t print any error status for any device.
If you’ve recently upgraded a driver, reboot the system.
If that doesn’t help, create a new virtual environment and install the latest stable release (or nightly binary).