PyTorchインストールメモ

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

公式ウェブサイト

• PyTorch

環境

• 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

終わり