最近、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 を実行します。