pytorch バージョンと torchvision バージョンの対応と torchvision インストール

   最近、torchvision をインストールすると必ず失敗し、torchvision のバージョンと torch のバージョンが一致していないことがわかりました.技術交流グループのマスターの指導を通じて、torchvision のバージョンと torch のバージョンが一致する関係にあることがわかりました.次に、マイニングの経験を次のように共有します。

    ①Torchvisionとpytorch版の対応説明アドレス:https://github.com/pytorch/vision

   ②torchvisionダウンロードアドレス:https://download.pytorch.org/whl/cu110/torch_stable.html

    pytorch のインストールについては、この記事では詳しく説明しませんが、詳しく知りたい場合は、私の別のブログ ( https://blog.csdn.net/weixin_41267342/article/details/112066981 ) を参照してください。トーチビジョンの取り付け。

 1. pytorch のバージョンを確認する方法:

輸入トーチ
print(torch.__version__)

 出力結果: 1.7.1+cu110、pytorch バージョンは 1.7.1、cu110 は GPU アクセラレーション コンピューティングのサポートを示し、gpu バージョンは: 11

2. オンラインで情報を確認し、次のように touchvision をインストールします。

    ①アナコンダ:

     conda install torchvision -c pytorch

    ②ピップ:

     pip インストール torchvision

   ③ソースより:

    python setup.py インストール
   # または、OSX の場合
   # MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py インストール

   pycharm のターミナルでコマンドを実行します: pip install torchvision  

   ただし、インストール プロセス中にエラーが報告されます。

        エラー: 要件 torch==1.4.0 を満たすバージョンが見つかりませんでした (torchvision から) (バージョン: 0.1.2、0.1.2.post1、0.1.2.post2、1.7.1 から) エラー: 一致するディストリビューションがありませ
       んtorch==1.4.0 で見つかりました (torchvision から)

    エラー メッセージは、torchvision のバージョンが torch のバージョンと一致しないというものです。

3. 次に、pytorch に一致する torvision バージョンを見つける方法を見つけます。

   インターネットで情報を確認したところ、1.5以下のバージョンが多く、私のpytorchのバージョンは1.7.1+cu110で、記事の最新バージョンを上回っています。

  次に、指示に一致する最新バージョンを見つける方法を見つけることです. これは github で入手できます. 具体的なアドレスは次のとおりです:

   https://github.com/pytorch/vision

最新の更新された公式 Web サイトの手順を確認できます。

私の 1.7.1 に一致する torchvison バージョンは 0.8.2 です

4. torchvison アドレスをダウンロードします: https://download.pytorch.org/whl/cu110/torch_stable.html

    torchvision-0.8.2%2Bcu110-cp38-cp38-win_amd64.whl での命名の説明:

    ① torchvision-0.8.2: torchvision のバージョンが 0.8.2 であることを示します

   ②cu110 :対応cuda(GPU)バージョン:11

   ③ cp38: anaconda の python のバージョンは python38 です。

   ④ win_amd64: 64 ビット Windows システム用。

     私のアナコンドの python バージョンは python38 で、windows64 ビット システムで実行されるため、必要なバージョンはtorchvision-0.8.2%2Bcu110-cp38-cp38-win_amd64.whlです。

5. ダウンロード後、指定した場所に配置します。例: F:\WorkSpace\python\project_one>、pycharm のターミナル ウィンドウから pip install torchvision-0.8.2+cu110-cp38-cp38-win_amd64.whl を実行します。

 

 

 

おすすめ

転載: blog.csdn.net/weixin_41267342/article/details/113878423