cuda9.1+cuDNN7.1+pytorch1.1-win10 での GPU のインストール

1. cuda ダウンロードのインストールと設定

1.1 コンピュータが cuda をインストールできるかどうかを確認します。以下のような方法:

打开控制面板,然后在右上方的搜索框里输入NVIDIA,如下图所示:第一张是打开控制面板时的图,第二张是输入NVIDIA之后,从图中的左上角可以看到搜索出来的NVIDIA。

ここに画像の説明を挿入
ここに画像の説明を挿入
前の手順を完了したら、次のインターフェイスに入ります。ここでは、下図の左側のメニューに示すように、NVIDIA とグラフィックス カードのドライバー バージョンに関する情報を表示できます。
ここに画像の説明を挿入
[ヘルプ] メニューをクリックし、下の図の赤いボックスに示すように、ドロップダウン メニューから [システム情報] オプションを選択します。
ここに画像の説明を挿入
ポップアップ システム情報ウィンドウには 2 つのメニュー ページがあります。[ディスプレイ] と [コンポーネント]、コンポーネントの選択、多くのファイル名が表示され、ファイル名で NVCUDA が見つかり、次の図の赤いボックスで示すように、製品名の列で NVCUDA のバージョンが表示されます。CUDA のバージョンは 9.1 です。
ここに画像の説明を挿入
もちろん、コンピュータがcudaをサポートしていない場合は、上記のように表示されません。

1.2 cudaのダウンロードと設定

ダウンロードする必要があるバージョンがわかったら、cuda 公式 Web サイトにアクセスしてダウンロードします。
ここに画像の説明を挿入
ここに画像の説明を挿入
ここではローカルを選択することをお勧めします。これにより、いくつかの関連パッケージを一度に解決し、将来の問題を回避できます。一番下には、インストールできるパッチがまだあります。これは、cuda のインストール後に実行できます。

次に、インストールプロセスがあり、ダブルクリックして一時解凍ディレクトリを開いて表示します。変更する必要はなく、デフォルトのままで問題ありません。

次に、NVIDIA インストール プロセスに入り、カスタム インストール方法を選択し、VS と表示用の 3 つのオプションのチェックを外しました。VS はすでにインストールされており、次のコンポーネントをインストールしたくないからです。CUDAのインストールディレクトリはデフォルトでCドライブにインストールできます。
ここに画像の説明を挿入
ここに画像の説明を挿入
インストールが完了すると、環境変数が設定されます。環境変数の設定を次の図に示します。
ここに画像の説明を挿入
cuda が正常にインストールされているかどうかをテストするには、win+R で cmd を開き、nvcc -V を入力するだけです。バージョン番号が表示されれば成功です。

2. CUDNN のダウンロードと設定

ここをクリックて公式サイトに入り、cudnnをダウンロードすることを選択しますが、ここではアカウントの登録が必要で、アンケートに回答するとダウンロードページに進むことができます。とにかくステップバイステップで実行できます。

前回のCUDAのバージョンは9.1でしたので、原則としてcuDNNとcudaのバージョンは同じである必要がありますが、その後cuDNNがアップデートされたため、対応するcuda9.1のバージョンはありません。そこで、cuda9.0のバージョンをダウン選択しました。
ここに画像の説明を挿入
ダウンロード後、解凍して、CUDNN 圧縮パッケージ内の bin、include、lib ファイルを CUDA インストール ディレクトリに直接コピーし、インストールを直接上書きします。
ここに画像の説明を挿入

3. pytorch-gpuのインストール

Pytorch の公式 Web サイトに直接アクセスします。最初は conda を使ってインストールしていたのですが、いつも conda のインストールが失敗してしまい、原因が分かったのは conda を使うと URL に reposd failed と表示されてしまうためで、これは公式サイトの問題のようです。その後、pip でインストールしてみましたが、うまくいきました。ただし、conda メソッドも実行可能ですが、それを決定できるかどうかはキャラクターによって異なります (現在、清華鏡はなくなっています!)。
ここに画像の説明を挿入
両方の矢印が使用可能で、上の矢印は conda、下の矢印は pip です。

次に、コマンド ラインは pytorch GPU バージョンが正常にインストールされているかどうかを確認し、インストールが成功したことを示す true を返します。バージョン番号も確認できます。
ここに画像の説明を挿入
ここに画像の説明を挿入

今だけ楽しんでください!

Je suppose que tu aimes

Origine blog.csdn.net/qq_42194665/article/details/104999962
conseillé
Classement