MXNet is an ultra-scalable deep learning framework. This version uses CUDA-10.2 and MKLDNN.
Apache 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.
For feature requests on the PyPI package, suggestions, and issue reports, create an issue by clicking here.
This package supports Linux and Windows platforms. You may also want to check: - mxnet-cu102 with CUDA-10.2 support. - mxnet-cu101 with CUDA-10.1 support. - mxnet-cu101mkl with CUDA-10.1 and MKLDNN support. - mxnet-cu100 with CUDA-10.0 support. - mxnet-cu100mkl with CUDA-10.0 support and MKLDNN support. - mxnet-cu92 with CUDA-9.2 support. - mxnet-cu92mkl with CUDA-9.2 support and MKLDNN support. - mxnet-cu90 with CUDA-9.0 support. - mxnet-cu90mkl with CUDA-9.0 support and MKLDNN support. - mxnet-cu80 with CUDA-8.0 support. - mxnet-cu80mkl with CUDA-8.0 support and MKLDNN support. - mxnet-cu75 with CUDA-7.5 support. - mxnet-cu75mkl with CUDA-7.5 support and MKLDNN support. - mxnet-mkl with MKLDNN support. - mxnet.
To install for other platforms (e.g. Windows, Raspberry Pi/ARM) or other versions of CUDA, check Installing MXNet for instructions on building from source.
pip install mxnet-cu102mkl
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Hashes for mxnet_cu102mkl-1.6.0.post0-py2.py3-none-manylinux1_x86_64.whl