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

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

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

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

先人に感謝.

サイト内の関連ページ:


前準備

OS のシステム更新

Ubuntu で 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 GPU を使うとき)NVIDIA グラフィックスボード・ドライバのインストール

GPU とは,グラフィックス・プロセッシング・ユニットの略で、コンピュータグラフィックス関連の機能,乗算や加算の並列処理の機能などがある.

ダウンロードページ

詳しくは,次のように別ページで説明している.インストール手順が分からなくなったり,不具合があったときは,次に書いている別ページの説明を確認すること.

(NVIDIA GPU を使うとき)NVIDIA CUDA ツールキットのインストール

CUDA とは,NVIDIA社が提供している GPU 用のプラットフォームである.

ダウンロードページ

TensorFlow 2.1 以上の GPU 版での,NVIDIA CUDA のバージョンは 10.1指定されている. TensorFlow 1.15の GPU 版での,NVIDIA CUDA のバージョンは 10.0指定されている. 指定されているバージョンより高いものは使わない. その根拠は次のページ. URL: https://www.tensorflow.org/install/source#common_installation_problems

詳細説明

(NVIDIA GPU を使うとき)NVIDIA cuDNN のインストール

ダウンロードページ

次のページには,TensorFlow 2.1 以上の GPU 版での,cuDNN のバージョンは 7.6.そして,TensorFlow 1.13 以上 TensorFlow 2.0 までの GPU 版での,cuDNN のバージョンは7.4 が指定されている.

URL: https://www.tensorflow.org/install/source#common_installation_problems

詳細説明


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 python3 -m pip install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-linux_x86_64.whl
    sudo python3 -m pip 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 プログラムを動かす.

    ※ Python プログラムを動かすために, 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())'
    


本サイトは金子邦彦研究室のWebページです.サイトマップは,サイトマップのページをご覧下さい. 本サイト内の検索は,サイト内検索のページをご利用下さい.

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