Vengineerの戯言

人生は短いけど、長いです。人生を楽しみましょう!

TensorFlow Runtimeに GPU backend が入った

@Vengineerの戯言 : Twitter
SystemVerilogの世界へようこそすべては、SystemC v0.9公開から始まった 

記録のために、

TensorFlow Runtime、今まで CPU だけでしたが、GPU (CUDA) backend が入りました

github.com

build --@rules_cuda//cuda:cuda_targets=sm_60

あって、まだまだ、そんなに対応していないっぽい。

cuDNN、cuSOLVER、BLASなどの wrapper file はあるっぽい。

HIP wrapper もある。