Running Google Colab notebooks on Framework Desktop (and utilizing the GPU)

In case anyone is interested, i created a google colab runtime image based on the google colab runtime that installs latest ROCm and pytorch versions and allows to run google colab notebooks on a ROCm compatible AMD GPU.

I am happily using it with some AI courses that are utilizing Google Colab. I can run the notebooks on the Framework Desktop and GPU is utilized.

1 Like

Thanks for sharing. I’ll definitely have a play with it when I get some time over the coming weeks as that’s been on my to-do list for a bit too long

1 Like

i’d be happy about any feedback :slight_smile: