Google Colab недавно представил новую среду выполнения с графическим процессором Tesla K80.

Эта среда выполнения основана на CUDA 8, в то время как для новейшей версии MXNet/Gluon с поддержкой графического процессора требуется CUDA 9.

В старых версиях дистрибутива MXNet, совместимого с CUDA 8 (mxnet-cu80), возникает исключение при загрузке драйвера CUDA, установленного в Google Colab. Требуется более новая версия.

Метку последней версии MXNet, совместимой с CUDA 8, можно выбрать здесь:



mxnet-cu80
MXNet — это сверхмасштабируемая среда глубокого обучения. В этой версии используется CUDA-8.0.pypi.org



На момент написания этого поста метка последней версии была «1.0.1b20180202».

В верхней части блокнота Google Colab, в котором используется MXNet/Gluon, запустите команду для установки библиотеки.

Теперь мы готовы идти :-)