0
2

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 5 years have passed since last update.

tensorflow-gpu 1.15.3 をWin10で動かす

Last updated at Posted at 2020-07-17

環境を整えようとして大苦戦したので、判明した導入方法を紹介します。
※Intel AVX2を使うのでIntelの第二世代Coreプロセッサー(Sandy Bridge)以前の古いCPUだとインストールできません。

python

3.5.x ~ 3.7.x を利用しましょう。

モジュールインストール

pip install -U pip
pip uninstall tensorflow
pip install tensorflow-gpu==1.15.3 tensorflow-estimator==1.15.1

CUDA インストール

CUDA Toolkit 10.0 をインストールしましょう。

cuDNN の展開

v7.6.5 (November 5th, 2019), for CUDA 10.0 をダウンロードしましょう。
https://developer.nvidia.com/rdp/cudnn-download
※ログインが必要なのでログイン画面の[CREATE AN ACCOUNT]からアカウント作成(無料)してください。
ダウンロードしたzipファイルを解凍し
中身をC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0にコピーして下さい。

PATHを通しましょう

これら3つのパスを環境変数に設定して下さい。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\CUPTI\libx64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include

以上で、Win10にてtensorflow-gpu 1.15.3が利用可能になります。
お疲れさまでした。

0
2
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
0
2

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?