TensorFlowとGPUの対応メモ

TensorFlowはNVIDIAのRTX20xx系GPUで動くのか。
少し調べたので「TensorFlowとGPUの対応メモ」として残しておく。

NVIDIAのGPUを使うにはCUDAが必要

以下Wikipediaにも書いてあるように、NVIDIAのGPUで計算するにはCUDAというソフトウェアを経由する必要がある。

NVIDIA製GPUにおいては、OpenCL/DirectComputeなどの類似APIコールは、すべて共通のGPGPUプラットフォームであるCUDAを経由することになる。
CUDA - Wikipedia

TensorFlow 1.x系はCUDA9まで

TensorFlow公式サイトを見るにTensorFlow 1.x系は CUDA9 までが対応している。
CUDA10 について書かれていないので、TensorFlow 1.x系は CUDA10 には対応していない。
Build from source on Windows  |  TensorFlow

TuringアーキテクチャはCUDA10から対応

NVIDIAのGPUであるGTX1080などの GTX10xx系 はPascalアーキテクチャ。
RTX2080などの RTX20xx系 はTuringアーキテクチャである。

そしてTuringアーキテクチャに対応したCUDAは CUDA10 からである。
CUDA - Wikipedia

結論

• TensorFlow 1.x系はCUDA9まで
• TuringアーキテクチャはCUDA10から対応
• RTX20xx系はTuringアーキテクチャなのでTensorFlow 1.x系は対応していない

猛者はソースからビルドして動かしている。
以前「GPUは今が買い時」と書いたが、最新版を買った為にソフト側の対応を待たなければならないケースもあるので、要注意である。

最近TensorFlow2.0のアルファ版が出てきたので、今後は2.0系が主流になるだろう。