keep-loving-pythonのブログ

Pythonを愛し続けたいです(Pythonが流行っている限りですが。。。)

torchとtorchvisionのversionの組合せは、pipによる依存性確認ではわからないというか、そもそも、決まってると思います。

torchをあるバージョンにしたい。torchvisionをあるバージョンにしたい。これは、バラバラには無理だと思います。

>python -m pipdeptree | grep torchvision

その結果、以下みたいな感じで、torchのことは出てきません。

・・・・
* torchtext==0.11.2
 - torch [required: ==1.10.2, installed: 1.8.1]
* tsfresh==0.12.0
 - pandas [required: >=0.20.3,<=0.23.4, installed: 1.3.5]
* typer==0.3.2
 - click [required: >=7.1.1,<7.2.0, installed: 8.1.3]
------------------------------------------------------------------------
Warning!! Cyclic dependencies found:
* Naked => Naked => Naked
------------------------------------------------------------------------
  - torchvision [required: >=0.8.2, installed: 0.9.1]
  - torchvision [required: Any, installed: 0.9.1]
  - torchvision [required: Any, installed: 0.9.1]


どう決まっているか。

pypi torchvision

https://pypi.org/project/torchvision/

pytorchのサイト

たぶん、こうすれば良い。 (すみません、この項、間違い!!です。)

例えば、torchを1.13.0に是非したいという場合には、そのように、pipでインストールすれば良い。
torchvisionは、勝手にそれにあったもの(バージョン)にかわるはず。

訂正(普通にPytorchのサイトから。。。)

訂正します。いろんな場合があるみたいで。。。 結局。

https://pytorch.org/get-started/previous-versions/

たとえば、↓みたいな感じで。。。

python -m pip install torch==1.13.1+cpu torchvision==0.14.1+cpu torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cpu

で、宜しく。