トップページ -> コンピュータ -> Ubuntu -> Python 2 で,PyTorch, Caffe2 のインストール(Ubuntu 18.04 上)
[サイトマップへ], [サイト内検索へ],

Python 2 で,PyTorch, Caffe2 のインストール(Ubuntu 18.04 上)

サイト構成 連絡先,業績など 実践知識 データの扱い コンピュータ 教材 サポートページ

ユースケース: Ubuntu の Python 2 (バージョン 2)で,PyTorch, Caffe2 を使いたい

次のページの記載に従い PyTorch(PyTorch には Caffe2 同封)をインストールする.

https://pytorch.org/get-started/locally/

先人に感謝.

サイト内の関連Webページ: PyTorch, Caffe2 のインストール(Ubuntu 18.04 上)


前準備

Ubuntu のシステム更新

端末で,次のコマンドを実行.

sudo apt update
sudo apt -yV upgrade
sudo shutdown -r now

Python2 開発用ファイル,セットアップツール,pip,numpy のインストール

端末で,次のコマンドを実行.

sudo apt -yV install python-dev python-setuptools python-pip python-numpy

NVIDIA グラフィックスカードの確認とドライバのインストール,NVIDIA CUDA ツールキット,NVIDIA cuDNN 再配布パッケージのインストール

Ubuntu での手順は,「別のページ」で説明している. NVIDIA グラフィックスカードがある場合のみ行うこと.


PyTorch のインストール

  1. Python のバージョンを確認しておく

    端末で,次のコマンドを実行. パージョンが「2.・・・」のように表示されれば OK.

    python
    

  2. 前提パッケージのインストール

    端末で,次のコマンドを実行.

    sudo apt install python-numpy
    sudo pip install -U protobuf
    

  3. PyTorch の「はじめよう」の Web ページを選ぶ

    https://pytorch.org/get-started/locally/

  4. 種類を選ぶ

    Linux, pip, Python 2.7 CUDA 10.0 の実行例

  5. 「Run the command」のところに,コマンドが表示されるので確認する

  6. 端末で,いま表示されたコマンドを実行

    pip3 の前には,sudo を付けて実行する

    Linux, pip, Python 2.7, CUDA 10.0 の実行例

    sudo pip install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp27-cp27mu-linux_x86_64.whl
    sudo pip install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp27-cp27mu-linux_x86_64.whl
    

  7. その結果,エラーメッセージが出ていないことを確認.

  8. PyTorch のバージョン確認

    端末で,次のコマンドを実行.

    ※ バージョン番号が表示されれば OK.

    python -c "import torch; print( torch.__version__ )"
    

  9. PyTorch の動作確認

    https://pytorch.org/get-started/locally/ に記載のサンプルプログラムを実行してみる

    python を起動し,次のPython プログラムを動かす.

    from __future__ import print_function
    import torch
    x = torch.rand(5, 3)
    print(x)
    exit()
    

  10. Caffe 2 の確認

    「Success」と表示されることを確認. 「Failure」と表示されるときは,このページで説明している「前提パッケージのインストール」を忘れていないかを確認すること.

    python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"
    

  11. Caffe 2 GPU 版の確認 (GPU版をインストールしたときのみ実行)

    CUDA デバイスの数が表示されることを確認

    python -c 'from caffe2.python import workspace; print(workspace.NumCudaDevices())'