GPU Support

Animius uses Tensorflow as its deep learning library. To enable GPU support on training and using your virtual assistant, you will need to install a GPU-support enabled version of Tensorflow (Read more here).

To simplify the installation of Animius and avoid library conflicts, we recommend using a Animius Docker image with GPU support based on the Tensorflow-GPU image (Linux only). This setup only requires the NVIDIA® GPU drivers.

For more instructions on Tensorflow-gpu installation, please refer to the Tensorflow guide. Animius will automatically use the version of Tensorflow you install (i.e. GPU if you install tensorflow-gpu and CPU if you install tensorflow).