PytorchをWindows7 64ビットシステムにインストールする(2020.12.27)

1.環境への備え

        特定の操作について、ブログWindows 7のセクション4.3.9を参照して、TensorFlowディープラーニングフレームワーク(2020.11.22)を構築してくださいデフォルトの環境はこのセクションに基づいています。

1.1コンピューターのGPUをチェックし、グラフィックカードドライバーがインストールされていること、およびCUDAの計算能力がサポートされていることを確認します

コンピューターGPU
グラフィックカードドライバがインストールされています
NVIDIAコントロールパネル
グラフィックカードCUDAコンピューティングパワー
        コンピューターのGPUグラフィックカードはGTX970であるため、**グラフィックカードドライバー**がインストールされていることを確認してください。[NVIDIA公式Webサイト](https://developer.nvidia.com/zh-cn/cuda-gpus)にアクセスできます。 )コンピュータグラフィックスカードを表示するには、CUDAの計算能力は5.2です。

1.2 CUDA、cudnn、およびVisual C ++生成ツール2019がインストールされていることを確認します

ここに画像の説明を挿入
        以前にtensorflow-gpu2.3.1をインストールするために、MSVC 2019–vc_redist.x64.exeCUDA 10.1cudnn7.6.5を選択しました。もちろん、GPUを使用して深層学習コードを実行するために、CUDAバージョンとcudnnバージョンは独自のグラフィックカードに応じてサポートされます。CUDAの計算能力を選択できます。
ここに画像の説明を挿入
ここに画像の説明を挿入

ここに画像の説明を挿入
        cuDNNをダウンロードして解凍した後、cudaフォルダーにlib \ x64を含む3つのフォルダーbinがあります。これらの3つのフォルダー(cudnn64_7.dll、cudnn.h、cudnn.lib)内のファイルをCUDAインストールディレクトリにコピーする必要があります。フォルダに対応します。

1.3 Condaがインストールされ、チャネルアドレスが清華ソースミラーリングで構成されていることを確認します

ここに画像の説明を挿入

2.Pytorchのインストール

        勝利+ Rは、CMDウィンドウを開き、次のコマンドを入力します。nvcc -VCUDAのバージョンとコンパイラドライバの表示
ここに画像の説明を挿入
        コマンドを入力していき、CMDを:conda create --n jjgpytorch python=3.6 //创建名称为jjgpytorch的虚拟环境
ここに画像の説明を挿入
        あなたが入力する必要がありますしPytorchのWHLパッケージのウェブサイトをダウンロードしcu101 /トーチ-1.4.0-をcp36-cp36m-win_amd64.whlファイル、Dにコピー:\ Anaconda3 \ envs \ jjgpytorch \ Scriptsフォルダー
ここに画像の説明を挿入
        のフォルダーウィンドウでcmdを開き、pip install torch-1.4.0-cp36-cp36m-win_amd64.whlPytorchをインストールするための直接入力の場合numpy.core.multiarrayが失敗するように求められることがありますインポートするため、numpyをインストールする必要があります。次にtorchをインストールします。

pip install numpy 
pip install torch-1.4.0-cp36-cp36m-win_amd64.whl

ここに画像の説明を挿入
        上の図では、torch-1.4.0を正常にインストールするように求められ、警告メッセージが表示され、2つのexeファイルが配置さているディレクトリD:\ Anaconda3 \ envs \ jjgpytorch \ Scriptsを配置する必要があることを示しています。システム環境変数Pathの下に、以下を追加します。
ここに画像の説明を挿入
ここに画像の説明を挿入
jjgpytorch環境にインストールされているパッケージを表示します。

conda activate jjgpytorch
pip list

ここに画像の説明を挿入

3.GPUバージョンのPytorchが正常にインストールされているかどうかをテストします

検証コード:

import torch
x = torch.rand(5,5)
print(x)
torch..cuda.is_available() #如果返回True,GPU版Pytorch成功安装完毕

3.1cmdウィンドウの操作

ここに画像の説明を挿入

3.2 JupyterNotebookの操作

ここに画像の説明を挿入

3.3スパイダーの操作

ここに画像の説明を挿入

3.4PyCharmの実行

ここに画像の説明を挿入

おすすめ

転載: blog.csdn.net/jing_zhong/article/details/111822940