go on colab, export the libraries you are using and their versions with:
pip freeze > requirements.txt

From there, pip install the libraries in the txt locally.
Clear GPU cache, re-run and it should work if its library related.