JossCamp

you need to install a specific version of pytorch > python -m pip install torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118 After that, it should work correctly.

Yes, it's possible. There shouldn't be any problems.

I don't know why in Colab, whenever something gets updated, it always has to mess up everything else. To solve this problem, you need to execute this command. > !python...

Google always ends up ruining something in each update, you need to use a specific version: `!pip install jax==0.4.19 jaxlib==0.4.19 -f https://storage.googleapis.com/jax-releases/jax_releases.html` This solves the problem for now.