MXNet is a deep learning framework designed for both efficiency and flexibility. This version only works with CUDA-8.0.
Project description
MXNet is a deep learning framework designed for both efficiency and flexibility. It allows you to mix the flavours of deep learning programs together to maximize the efficiency and your productivity.
Prerequisites
This package supports Linux platform only, and requires that CUDA-8.0 and cuDNN-5.1 are already installed, along with the proper NVIDIA driver. For package with CUDA-7.5 support on Linux, check mxnet-cu75.
To download, check CUDA download and cuDNN download pages. For more instructions, check CUDA Toolkit online documentation.
To install for other platforms (e.g. Windows) or other versions of CUDA or cuDNN, check Build Instruction.
Installation
To install:
pip install mxnet-cu80
