Tensorflowの導入方法の説明
Tensorflowの導入はTensorflowのバージョンとCUDAのバージョンを合わせることが重要です。
バージョン対応表
https://www.tensorflow.org/install/source?hl=en#tested_build_configurations
今回は以下の組み合わせで導入します。
- CUDA:10.0
- cuDNN v7.6.2 (July 22, 2019), for CUDA 10.0
- Tensorflow-gpu 1.14.0
CUDAのダウンロード。旧バージョンのダウンロードは以下から行える(2019/8時点)
https://developer.nvidia.com/cuda-toolkit-archive
CuDNNのダウンロード。アカウントが必要
https://developer.nvidia.com/rdp/form/cudnn-download-survey
Tensorflowのインストール
pip install tensorflow-gpu==1.14.0