Windows7x64システムでのPytorchのインストール
1.環境への備え
特定の操作については、ブログWindows 7のセクション4.3.9を参照して、TensorFlowディープラーニングフレームワーク(2020.11.22)を構築してください。デフォルトの環境はこのセクションに基づいています。
1.1コンピューターのGPUをチェックし、グラフィックカードドライバーがインストールされていること、およびCUDAの計算能力がサポートされていることを確認します
|
|
|
|
1.2 CUDA、cudnn、およびVisual C ++生成ツール2019がインストールされていることを確認します
以前にtensorflow-gpu2.3.1をインストールするために、MSVC 2019–vc_redist.x64.exe、CUDA 10.1、cudnn7.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 -V
CUDAのバージョンとコンパイラドライバの表示
コマンドを入力していき、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.whl
Pytorchをインストールするための直接入力の場合、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成功安装完毕