windows10でtensorFlow

2018年11月30日 19:59

環境
windows10
GTX1070
Anaconda 5.3.1
CUDA 10

anacondaをインストール
https://www.anaconda.com/download/#windows
3.xバージョンを入れる。

GPUなら

・CUDA toolkit 10 をダウンロード、インストール
https://developer.nvidia.com/cuda-downloads

・cuDNN 10 ダウンロード
https://developer.nvidia.com/rdp/cudnn-download
登録が必要。めんどくさい。

・解凍したものをCUDA内の \bin, \include, \lib に突っ込む
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0のなかにそれぞれのフォルダがある。

・conda内 enviromentタブからCreateで環境構築「tf-gpu」 pythonのverは3.6に

・condaから tensorflow-gpuをinstall

以下のコードのpythonファイルを作ってcondaの環境からOpenTerminalして走らせてみる。
>> python xxxxxx.py

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

・conda内 enviromentタブからCreateで環境構築「tf」 pythonのverは3.6に

・condaから tensorflowをinstall

次はOpenPoseを動かしてみる。
https://note.mu/uneori/n/n337b11bdfa3b