Auto ML library made by mxnet. https://autogluon.mxnet.io/
Setup on Linux From the official page
# Here we assume CUDA 10.0 is installed. You should change the number
# according to your own CUDA version (e.g. mxnet-cu101 for CUDA 10.1).
pip install --upgrade mxnet-cu100
pip install autogluon
pip install --upgrade mxnet-cu100
pip install autogluon
from autogluon import TabularPrediction as task
ImportError Traceback (most recent call last)
in () 1 get_ipython().system('pip install mxnet autogluon') ----> 2 from autogluon import TabularPrediction as task
Get angry. The same applies when using the CPU.
After trying various things, it was solved by the following method. https://github.com/awslabs/autogluon/issues/163
!pip uninstall -y mkl
!pip install --upgrade mxnet-cu100
!pip install autogluon
!pip install -U ipykernel
# and restart runtime
It may fail again if the GPU or cuda of colab changes in the future.
Recommended Posts