Prerequisites

This package supports Linux and Windows platforms. You may also want to check:

To download CUDA, check CUDA download. For more instructions, check CUDA Toolkit online documentation.

To use this package on Linux you need the libquadmath.so.0 shared library. On Debian based systems, including Ubuntu, run sudo apt install libquadmath0 to install the shared library. On RHEL based systems, including CentOS, run sudo yum install libquadmath to install the shared library. As libquadmath.so.0 is a GPL library and MXNet part of the Apache Software Foundation, MXNet must not redistribute libquadmath.so.0 as part of the Pypi package and users must manually install it.

To install for other platforms (e.g. Windows, Raspberry Pi/ARM) or other versions, check Installing MXNet for instructions on building from source.

Installation

To install:

pip install mxnet-cu101