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

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

サイト構成 連絡先,業績など コンピュータ実習 データの扱い コンピュータ活用 教材(公開) サポートページ

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

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

先人に感謝.

サイト内の関連Webページ:


前準備

Ubuntu のシステム更新

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

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

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

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

sudo apt -yV install python3-dev python3-setuptools python3-pip 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 プログラムを動かしたい. そのために,「Python コンソール」を使う.

    PyCharmspyder を使うのが簡単.

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

  9. Caffe 2 の確認

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

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

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

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

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