WindowsプラットフォームのインストールTensorFlowのGPU

逐語https://www.dazhuanlan.com/2019/08/26/5d63016142d97/

国際的な慣行は、公式ドキュメントを参照してください。公式の推奨pipインストールを、あなたはTensorFlowをインストールする前にインストールする必要があります

CUDA 10.1
cuDNN 7.6
Microsoft Visual C++ 2015 Redistributable Update 3

そして、実行

pip install tensorflow-gpu

インストール。

テストのインストールは成功しました：

import tensorflow as tf

エラー、不足しているDLLファイルを示唆しています。編集者注ビットを見つけ、ほとんどの人は、この方法は、インストールすることで解決しVC++ 2015 Redistributable Update 3、再インストールVC++ 2015 Update 3アンインストールした後TensorFlow-GPU、次にインストール、または同じ過ちを報告しました。

他の取付けCondaで

ふりPyTorch便利な公式インストールマニュアルを参照してくださいCondaインストール方法を、あなたにはチャンスを取るTensorFlow試み。

# 安装 PyTorch
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
# 安装 TensorFlow
conda install tensorflow-gpu

インストールのテスト後：

import tensorflow as tf
import torch

エラーなし、インストールの成功。