WindowsでPyTorchをインストールする

公式のホームページを見に行こう。そうすると今ブラウザを開いているPCで適切なインストール方法を教えてくれる。

多分これでいける。

!pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html
Looking in links: https://download.pytorch.org/whl/torch_stable.html
Collecting torch===1.4.0
  Downloading https://download.pytorch.org/whl/cu101/torch-1.4.0-cp36-cp36m-win_amd64.whl (796.8MB)
Collecting torchvision===0.5.0
  Downloading https://files.pythonhosted.org/packages/80/59/ba66ebe892ec8302048434455fb6a9944eb0a78315184f8757094ba95435/torchvision-0.5.0-cp36-cp36m-win_amd64.whl (1.2MB)
Collecting pillow>=4.1.1
  Downloading https://files.pythonhosted.org/packages/44/4b/78226761e8ce14686fa7fa834c357d5b0a933d7485955b716fab3071f7e5/Pillow-7.0.0-cp36-cp36m-win_amd64.whl (2.0MB)
Requirement already satisfied: numpy in c:\users\katsunari.murase\appdata\local\continuum\anaconda3\envs\myenv\lib\site-packages (from torchvision===0.5.0) (1.17.4)
Requirement already satisfied: six in c:\users\katsunari.murase\appdata\local\continuum\anaconda3\envs\myenv\lib\site-packages (from torchvision===0.5.0) (1.13.0)
Installing collected packages: torch, pillow, torchvision
Successfully installed pillow-7.0.0 torch-1.4.0 torchvision-0.5.0
WARNING: You are using pip version 19.3.1; however, version 20.0.2 is available.
You should consider upgrading via the 'python -m pip install --upgrade pip' command.

試しに使ってみる

import torch

適当な乱数を発生させて、5×3の行列を作成してみる。

x = torch.rand(5, 3)
print(x)
tensor([[0.6201, 0.8497, 0.9403],
        [0.8235, 0.3304, 0.6796],
        [0.6215, 0.1080, 0.9779],
        [0.9894, 0.3692, 0.5647],
        [0.8888, 0.7965, 0.4367]])

GPUが使える環境かどうかは以下のコマンドを実行するとわかる。

torch.cuda.is_available()
False