MXNet(Apache MXNet)は、柔軟性と高性能を兼ね備えた深層学習フレームワークです。ここでは、MXNetのインストール方法について、pip, conda, および Docker を用いた主要な手法に分けて詳しく説明します。
1. pipによるインストール
pip はPythonの標準的なパッケージ管理ツールであり、最も手軽にMXNetをインストールできる方法です。
CPU版のインストール:
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 コマンドで仮想環境を整備しつつインストールするのが便利です。
手順:
GPU版(CUDA 10.1など):
※ 利用可能なバージョンは conda search -c conda-forge mxnet* で確認できます。
3. Dockerによるインストール
Dockerを使用すると、環境構築の煩雑さを回避しつつ、コンテナベースでMXNetを利用できます。
CPU版コンテナ起動:
GPU版コンテナ起動(NVIDIA Dockerが必要):
または、Docker Hub上のMXNet公式イメージ:
※ GPU利用のためには nvidia-docker2 と適切なドライバのインストールが前提です。
4. 確認
インストール後、以下のPythonコードでMXNetが正しくインポートできるか確認します。
補足事項
-
MXNetのバージョンは最新でなくても安定版を使うのが望ましい場合があります。
-
GPU環境を利用する場合は、CUDAの互換性に十分注意してください。
生成日:2025/05/23