Qiita Teams that are logged in
You are not logged in to any team

Log in to Qiita Team
Community
OrganizationAdvent CalendarQiitadon (β)
Service
Qiita JobsQiita ZineQiita Blog
1
Help us understand the problem. What is going on with this article?
@yohiro

tensorflow-gpu導入メモ

はじめに

, , , でsklearnやsvmをかじってみたので、今度はTensorFlowをかじってみようと思い、その導入メモ。幸い、GPU搭載パソコンを所有しているため、GPU版のtensorflowを導入してみる。

参考教材:【画像判定AIアプリ開発・パート1】TensorFlow・Python・Flaskで作る画像判定AIアプリ開発入門

自分の環境

  • Windows 10
  • Anaconda3インストール済
  • GeForce GTX 1060搭載

導入要件

https://www.tensorflow.org/install/gpu?hl=ja
公式より、tensorflowに必要な要件は以下の通り
cap3.png

GPUドライバさえ揃えてしまえば口述のコマンド一発でインストールできるっぽい。
ので、まずはドライバをアップデート。

GPUドライバアップデート

まずはデバイスマネージャを起動し、「ディスプレイアダプター」の対象となるGPUを右クリ→「ドライバーの更新」をクリック
cap3.png

「ドライバー ソフトウェアの最新版を自動検索」を押せば後は自動でアップデートしてくれる。

tensorflow-gpuのインストール

Anacondaの環境から以下のコマンドを実行

conda install tensorflow-gpu

インストールの確認

Anacondaの環境から

python

と入力した後、

import tensorflow

と入力する。そうすると

2020-04-15 18:35:41.872948: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll

と表示され、インストール成功してるっぽい模様。

はまったところ

教材に従ってCUDAやcuDNNなどを個別にインストールしようとしていたが、CUDAをインストールしようとしたところ下記ページと同様の「Visual Studio Integrationで失敗」という症状になりインストールできなかった。

Windows端末にCUDAをインストールする際に、「Visual Studio Integration」のインストールに失敗する場合

結局、先述のコマンドでtensorflow-gpuに必要なものを自動で構成してくれるらしいので、CUDAやcuDNNの個別インストールは未実施。

1
Help us understand the problem. What is going on with this article?
Why not register and get more from Qiita?
  1. We will deliver articles that match you
    By following users and tags, you can catch up information on technical fields that you are interested in as a whole
  2. you can read useful information later efficiently
    By "stocking" the articles you like, you can search right away
yohiro
広島在住の組込系ソフトウェアエンジニアです。 ずっとCの人間でしたが、最近はPythonなど流行りモノに興味有り。

Comments

No comments
Sign up for free and join this conversation.
Sign Up
If you already have a Qiita account Login
1
Help us understand the problem. What is going on with this article?