windows10でtensorFlow

環境
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()

CPUなら

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

・condaから tensorflowをinstall

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


この記事が気に入ったらサポートをしてみませんか?