PyTorchのインストール(1.9.1 Windows版)

IT

これまでKeras/Tensorflowしか触っていなかったのだが、Kaggle等をみてるいるとPyTorchも少しわかっていたほうがよいと思い、家のPCにPyTorchをインストールした。

pytorch.org

最初に、condaとpipとどちらでインストールするか悩む。一応、家のPC環境はcondaベースなのだが、Keras/Tensorflowなどcondaでは最新が入れられず、pipでインストールしているので混在環境になっている。

guppi524.hatenablog.jpで書いているとおり、このPCにはCUDA 11.4がすでにインストールされているので、 stackoverflow.com

あたりをみつつ、pipでインストールすることとした。

(base) D:\Users\hoge>pip3 install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio===0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
Looking in links: https://download.pytorch.org/whl/torch_stable.html
Collecting torch==1.9.1+cu111
  Downloading https://download.pytorch.org/whl/cu111/torch-1.9.1%2Bcu111-cp38-cp38-win_amd64.whl (3128.1 MB)
     |████████████████████████████████| 3128.1 MB 190 bytes/s
Requirement already satisfied: typing-extensions in d:\users\tkonishi\miniconda3\lib\site-packages (from torch==1.9.1+cu111) (3.10.0.2)
Collecting torchaudio===0.9.1
  Downloading https://download.pytorch.org/whl/torchaudio-0.9.1-cp38-cp38-win_amd64.whl (216 kB)
     |████████████████████████████████| 216 kB 1.7 MB/s
Collecting torchvision==0.10.1+cu111
  Downloading https://download.pytorch.org/whl/cu111/torchvision-0.10.1%2Bcu111-cp38-cp38-win_amd64.whl (2.5 MB)
     |████████████████████████████████| 2.5 MB 3.2 MB/s
Requirement already satisfied: numpy in d:\users\tkonishi\miniconda3\lib\site-packages (from torchvision==0.10.1+cu111) (1.19.5)
Requirement already satisfied: pillow>=5.3.0 in d:\users\tkonishi\miniconda3\lib\site-packages (from torchvision==0.10.1+cu111) (8.0.1)
Installing collected packages: torch, torchvision, torchaudio
Successfully installed torch-1.9.1+cu111 torchaudio-0.9.1 torchvision-0.10.1+cu111

インストールパッケージの大きさは3GBくらい。家のネット環境だとそこそこ時間がかかった。

(base) D:\Users\hoge>conda list | grep torch
torch                     1.9.1+cu111              pypi_0    pypi
torchaudio                0.9.1                    pypi_0    pypi
torchvision               0.10.1+cu111             pypi_0    pypi
(base) D:\Users\hoge>python
Python 3.8.5 (default, Sep  3 2020, 21:29:08) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True
>>> quit()

GPUも使えそう。

以下の本を買ったので、しばらく勉強してみる予定です。

www.amazon.co.jp

  • 追記

上の本の序章のnotebookを動かすのに、

pip install japanize_matplotlib
pip install torchsummary
pip install torchviz

が必要でした。