Archai requires Python 3.6+ and PyTorch 1.2+. To install Python we highly recommend Anaconda. Archai works both on Linux as well as Windows.
We recommend installing from the source code:
git clone https://github.com/microsoft/archai.git
cd archai
install.sh # on Windows, use install.bat
You can also use dockers in the dockers folder. These are useful for large scale experimentation on compute clusters.
Vast majority of Archai code base works simply by installing the package by running the command pip install -e .
in archai source directory. However, this won't install pydot and graphviz to visualize the generated architectures. Also, if you have older pickle5 versions, you might get some errors which can be resolved by installing pickle5 using conda. This is what install script does.