インストール方法(pip、conda、Dockerなど)

MXNet(Apache MXNet)は、柔軟性と高性能を兼ね備えた深層学習フレームワークです。ここでは、MXNetのインストール方法について、pip, conda, および Docker を用いた主要な手法に分けて詳しく説明します。


1. pipによるインストール

pip はPythonの標準的なパッケージ管理ツールであり、最も手軽にMXNetをインストールできる方法です。

CPU版のインストール:

bash
pip install mxnet

GPU版のインストール:

使用しているCUDAバージョンに応じて以下のようにインストールします。

CUDA バージョン インストールコマンド
CUDA 11.0 pip install mxnet-cu110
CUDA 10.2 pip install mxnet-cu102
CUDA 10.1 pip install mxnet-cu101

※ GPU版を利用するには、対応するCUDAとcuDNNが事前にインストールされている必要があります。


2. condaによるインストール

Anaconda環境を使用している場合は、conda コマンドで仮想環境を整備しつつインストールするのが便利です。

手順:

bash
conda create -n mxnet_env python=3.9 conda activate mxnet_env conda install -c conda-forge mxnet

GPU版(CUDA 10.1など):

bash
conda install -c conda-forge mxnet-cu101

※ 利用可能なバージョンは conda search -c conda-forge mxnet* で確認できます。


3. Dockerによるインストール

Dockerを使用すると、環境構築の煩雑さを回避しつつ、コンテナベースでMXNetを利用できます。

CPU版コンテナ起動:

bash
docker run -it mxnet/python:latest

GPU版コンテナ起動(NVIDIA Dockerが必要):

bash
docker run --gpus all -it mxnet/python:latest_gpu

または、Docker Hub上のMXNet公式イメージ:

※ GPU利用のためには nvidia-docker2 と適切なドライバのインストールが前提です。


4. 確認

インストール後、以下のPythonコードでMXNetが正しくインポートできるか確認します。

python
import mxnet as mx a = mx.nd.array([1, 2, 3]) print(a)

補足事項

  • MXNetのバージョンは最新でなくても安定版を使うのが望ましい場合があります。

  • GPU環境を利用する場合は、CUDAの互換性に十分注意してください。

生成日:2025/05/23