AMD Radeon GPUでTensorFlowを動かす - tensorflow-directml編 - Qiita

はじめに TensorFlowの公式では、CUDAベース、つまりNVIDIAのGPU向けでの利用が記載されており、AMD GPU向けとはなっていない。しかし、AMDであろうとせっかくGPUがあるのだから機械学習に使ってみたい！ということで今回はtensorflow-directmlの紹介です。 AMDとしても機械学習利用を広めていきたいということで、ROCmライブラリーというのを出している（TensorFlowとPyTorchが使える）。 https://www.amd.com/ja/graphics/servers-solutions-rocm-ml （正直、あまりメジャーではない印象ですが、実際どうなのでしょうか。。。今回はこちらの紹介ではありません。） tensorflow-directml とは さて、本題の背景です。Microsoftとしても機械学習を推進しており、その中にはDi