LoginSignup
2
2

More than 3 years have passed since last update.

pytorchのインストール

Last updated at Posted at 2020-04-13

pytorchのインストールコマンド

pytorchのインストールに少し手間取ったのでメモを残します.
環境は以下のようになります.
OS:Windows10
Python3.6.6
Pytorch==1.1.0
Torchvision==0.3.0

単純に以下のように入力すればいいかと思いきや

pip install torch
pip install torchvision

tools.nnwrapのModuleNotFoundErrorが出てインストールできずバージョン指定も意味がなかった.

うまくいったコマンドはこちらです.

pip install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl
pip install https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp36-cp36m-win_amd64.whl
2
2
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
2
2