トップページ -> インストール,設定,利用 -> Ubuntu, Debian ソフトウエア -> PyTorch 1.3, Caffe2 のインストール(Ubuntu 上)
[サイトマップへ], [サイト内検索へ]

PyTorch 1.3, Caffe2 のインストール(Ubuntu 上)

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

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

先人に感謝.

サイト内の関連ページ:


前準備

OS のシステム更新

Ubuntu, Debian で OS のシステム更新を行うときは, 端末で,次のコマンドを実行.

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

Python3 開発用ファイル,pip, setuptools,numpy のインストール

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

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

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

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


PyTorch のインストール

  1. Python 3 のバージョンを調べる

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

    python3
    
  2. PyTorch の「はじめよう」の Web ページを開く

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

  3. 種類を選ぶ

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

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

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

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

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

    sudo pip3 install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-linux_x86_64.whl
    sudo pip3 install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp36-cp36m-linux_x86_64.whl
    
  6. その結果,エラーメッセージが出ていないことを確認.

  7. PyTorch のバージョン確認

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

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

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

  8. PyTorch の動作確認

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

    Python プログラムを動かす.

    ※ そのために, Windows では,「python」コマンドやPythonランチャーである「py」を使う. Ubuntu では「python3」コマンドを使う. あるいは, PyCharmなどにある Python コンソールも便利である.

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

  9. GPU が動作しているか確認したい場合

    import torch
    torch.cuda.is_available()
    
  10. Caffe 2 の確認

    「Success」と表示されることを確認

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

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

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

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


本サイトのサイトマップは,サイトマップのページをご覧下さい. 本サイト内の検索は,サイト内検索のページをご利用下さい.

問い合わせ先: 金子邦彦(かねこ くにひこ) [image]