1. 元の torch バージョンをアンインストールする
cmd で次のコマンドを入力できます
pip アンインストールトーチ
または、環境内の Lib\site-packages ファイルに直接移動して、torch (torchversion を含む) を含むすべてのファイルを手動で削除できます。
2. Pytorch 公式 Web サイトから GPU バージョンの torch をダウンロードします。
Pytorch 公式ウェブサイト アドレス: Pytorch 公式ウェブサイト
公式サイトに入ったら、下図のように設定します
次に、赤い線でマークされた Web サイトに入り、対応する GPU バージョンをダウンロードします。
3. 対応するバージョンを見つける
cmd に次のコマンドを入力します。
pip デバッグ --verbose
対応する torch および torchversion whl ファイルをダウンロードします
4. pip インストール用の whl ファイルをインストールしたディレクトリを入力します
同時に torchv-xxx.whl をインストールしてインストールを完了します
5. インストールが成功したかどうかを確認します