[work] Installing MXNet

淩亂°似流年 2022-05-08 14:58 210阅读 0赞

# Installing MXNet[¶][Link 1] #

Indicate your preferred configuration. Then, follow the customized commands to install MXNet.

v1.3.0

LinuxMacOSWindowsCloudDevices

PythonScalaRJuliaPerlCpp

CPUGPU

PipDockerBuild from Source

--------------------

$ pip install mxnet-cu92

--------------------

MXNet offers MKL pip packages that will be much faster when running on Intel hardware. Check the chart below for other options, refer to [PyPI for other MXNet pip packages][], or [validate your MXNet installation][].

![pip packages][]

**NOTES:**

*mxnet-cu92mkl* means the package is built with CUDA/cuDNN and MKL-DNN enabled and the CUDA version is 9.2.

All MKL pip packages are experimental prior to version 1.3.0.

CUDA should be installed first. Instructions can be found in the [CUDA dependencies section of the MXNet Ubuntu installation guide][].

**Important:** Make sure your installed CUDA version matches the CUDA version in the pip package. Check your CUDA version with the following command:

nvcc --version

You can either upgrade your CUDA install or install the MXNet package that supports your CUDA version.

--------------------

For more installation options, refer to the [MXNet Ubuntu installation guide][].

--------------------

# Source Download #

[Download][] your required version of MXNet and [build from source][].

[Link 1]: https://mxnet.incubator.apache.org/install/index.html?platform=Linux&language=Python&processor=GPU#installing-mxnet
[PyPI for other MXNet pip packages]: https://pypi.org/project/mxnet/
[validate your MXNet installation]: https://mxnet.incubator.apache.org/install/validate_mxnet.html
[pip packages]: https://raw.githubusercontent.com/dmlc/web-data/master/mxnet/install/pip-packages-1.3.0.png
[CUDA dependencies section of the MXNet Ubuntu installation guide]: https://mxnet.incubator.apache.org/install/ubuntu_setup.html#cuda-dependencies
[MXNet Ubuntu installation guide]: https://mxnet.incubator.apache.org/install/ubuntu_setup.html
[Download]: https://mxnet.incubator.apache.org/install/download.html
[build from source]: https://mxnet.incubator.apache.org/install/build_from_source.html