PyTorchインストールメモ

CycleGANを試そうとしたらPyTorchという深層学習フレームワークが必要なのでPyTorch1.0をインストールしたときのメモ。

公式ウェブサイト

環境

  • Ubuntu16.04
  • Python3.5
  • CUDA9.2

インストール

  • PyTorchのウェブサイトへ行き、自分の環境を選択するとRun the Comandに実行するコマンドが表示されるので実行する。私の場合は以下のコマンドであった。
  • pip3 install torch torchvision

検証

  • 以下のコードを実行して、うまくインストールできたか検証する。ここでは、ファイル名をtest.pyとする。
  •  
    from __future__ import print_function
    import torch
    x = torch.rand(5, 3)
    print(x)
    y = torch.cuda.is_available()
    print('cuda is available:', y)
    
  • 次のコマンドを実行し、うまくいった場合は5×3のランダムテンソルを表示し、Cudaが使えるときはTrue、使えないときはFalseを表示する。インストールはとても簡単。
    • python3  test.py

終わり

コメントを残す

メールアドレスが公開されることはありません。

このサイトはスパムを低減するために Akismet を使っています。コメントデータの処理方法の詳細はこちらをご覧ください