It could potentially be related. Could you post an executable code snippet, which raises the initial error using this module?



 agt:

If I’m in Google Colab, can I just put !CUDA_LAUNCH_BLOCKING=1 at the beginning of the cell?


It could work, if you set this environment variable before importing any other library, which might initialize the CUDA context. As it’s often not straightforward to use it properly in a Jupyter notebook, I usually recommend to run it in a terminal instead.