Not able to install Pytorch 2.0 on Google Colab

[TNitro]

Hey there. I’m new to this. I’m using google colab, I was just ignoring xformers till now, since there is this issue if you try to install xformers:
RuntimeError: Detected that PyTorch and torchvision were compiled with different CUDA versions. PyTorch has CUDA Version=11.7 and torchvision has CUDA Version=11.8. Please reinstall the torchvision that matches your PyTorch install.

but I would like to use Pytorch 2.0 so I can get better performance in Stable Diffusion.
Can someone help me install it the same way as this command does, but with pytorch 2.0?

!pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 torchtext==0.14.1 torchdata==0.5.1 --extra-index-url https://download.pytorch.org/whl/cu116 -U

[marksaroufim]

The default version on Colab is now 2.0 with cuda 11.8

So you shouldn’t need to install anything new

[TNitro]

but like I’ve said, this is the error I’m getting when starting it without that command line
RuntimeError: Detected that PyTorch and torchvision were compiled with different CUDA versions. PyTorch has CUDA Version=11.7 and torchvision has CUDA Version=11.8. Please reinstall the torchvision that matches your PyTorch install
I know its 2.0 that should be on google colab, but it doesn’t want to start, but instead gives that error message and shuts down my colab