Win10 は Tsinghua ソースを使用して pytorch-GPU バージョンをすばやくインストールします (推奨)

Cudaがインストールされているかどうかを確認してください

アナコンダプロンプトに入力

nvcc -V

上記の表示はインストールが完了したことを示します。

清華公園のダウンロード環境を構成する

アナコンダプロンプトにも入力してください

conda config --チャネルを追加 https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ 
conda config --チャネルを追加 https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ 
conda config --チャンネルを追加 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add チャネル https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ 
conda config --チャンネルを追加 https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --add チャネル https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

次に、C:\Users\Administrator ディレクトリ内の .condarc ファイルを削除し、ファイルを開いて defaults 行を削除します。次の図に示すように:

クリックしてpytorchの公式サイトにアクセスしてください

公式ウェブサイトリンク: https://pytorch.org/get-started/locally/。

次のように、独自のバージョンを選択します。

conda インストール pytorch torchvision cudatoolkit=10.1 -c pytorch

このコード行の意味は、pytorch をインストールすると torchvision もインストールされるため、後で torchvision をインストールする必要はありません。 cudatoolkit=10.1 は cuda バージョン 10.1 を意味します。 -c は公式ウェブサイトからダウンロードすることを意味します。 -cpu がない場合、GPU バージョンを示します。

pytorchのcudaバージョンを独自に設定する、cudaバージョンが低すぎる場合はどうすればよいか

上で述べたように、cudatoolkit=10.1 を自分の cuda バージョンに変更するだけです。私の場合は cuda 9.0 です。

conda インストール pytorch torchvision cudatoolkit=9.0 pytorch

インストールするだけです。

指定されたバージョンのtorchvisionをインストールする

コードを確認する際、作成者が特定のバージョンの torch または torchvision を要求する場合があります。これは、公式 Web サイトからファイルの以前のバージョンをダウンロードすることによってのみ実行できます。

以前のPyTorchバージョンをクリック

探しているバージョンに対応するファイルは上記にあります。ダウンロードしてpip installだけです。注意する必要があるバージョンの問題が多数あります。たとえば、cuda9.0 の torchvison はバージョン 0.3.0 のみをサポートします。

PyTorchがインストールされていることを確認する

それでおしまい！

パイソン
輸入トーチ
出口（）

インストール成功おめでとうございます。

要約する

上記は、Tsinghua ソースを使用して Windows 10 に pytorch-GPU バージョンをすばやくインストールするためのエディターの紹介です。皆様のお役に立てれば幸いです。ご質問がある場合は、メッセージを残してください。エディターがすぐに返信します。また、123WORDPRESS.COM ウェブサイトをサポートしてくださっている皆様にも感謝申し上げます。
この記事が役に立ったと思われた方は、ぜひ転載していただき、出典を明記してください。ありがとうございます!