Google Colab недавно представил новую среду выполнения с графическим процессором Tesla K80.
Эта среда выполнения основана на CUDA 8, в то время как для новейшей версии MXNet/Gluon с поддержкой графического процессора требуется CUDA 9.
В старых версиях дистрибутива MXNet, совместимого с CUDA 8 (mxnet-cu80), возникает исключение при загрузке драйвера CUDA, установленного в Google Colab. Требуется более новая версия.
Метку последней версии MXNet, совместимой с CUDA 8, можно выбрать здесь:
На момент написания этого поста метка последней версии была «1.0.1b20180202».
В верхней части блокнота Google Colab, в котором используется MXNet/Gluon, запустите команду для установки библиотеки.
Теперь мы готовы идти :-)