colab is broken

[eyaler]

colab is broken due to google removing cuda 10.1 here

this can be fixed by:

!apt-get -o Dpkg::Options::="--force-overwrite" install cuda-10-1 cuda-drivers
import os
os.environ["CUDA_HOME"] = '/usr/local/cuda-10.1'

however if you want to support modern GPUs as A100 you need cuda 11:

!apt-get -o Dpkg::Options::="--force-overwrite" install cuda-11-1
import os
cuda_home = '/usr/local/cuda-11.1'
os.environ["CUDA_HOME"] = cuda_home
with open(os.path.join(cuda_home, 'version.txt'), 'w') as f:
  f.write('CUDA Version 11.1')

I couldn't figure out how to install cudnn, but it works. see e.g. my accessible colab: https://colab.research.google.com/github/eyaler/avatars4all/blob/master/ganivut.ipynb

[Lolimipsu]

Thank you for raising this issue @eyaler it was very helpful. this should be pinned or at least put into the colab.