漢なら, 水冷 TITAN V で TensorFlow で機械学習したいですよね!
やりましょう!
とりあえず TITAN V 水冷化します.
TITAN V, 水冷化運用で 45 度あたりで安定して動いていて素晴らしいので水冷 TITAN V 16,777,216 枚くらい調達したい. 日々精進あるのみですね, 今日も一日ありがとうございます. pic.twitter.com/3ugdI1kJjJ
— Syoyo Fujita (@syoyo) May 29, 2018
すべては 優秀な GPU 水冷化若人さまのおかげです. ありがとうございます.
Tensorflow をコンパイルします.
Tensorflow は CUDA 9.1 には公式には未対応です. ドライバのバグ?
Issue は close されていますが, r1.9 でも prebuilt package は 9.0 になっています.
CUDA 9.0 でもいいのですが Volta 系の機能は CUDA 9.1 から入っているっぽいので, CUDA 9.1 でビルドしておきます.
このあたりを参考にしてビルドします.
- Ubuntu 16.04
- CUDA 9.1
- cuDNN 7.0
でビルドしました.
Benchmark
CIFAR10 でベンチマークしてみます.
ねんがんの TITAN V での tensorflow r1.9 CIFAR10 学習の実行にせいこうしたぞ!!!! 🎉🎉🎉🎉🎉🎉🎉㊗️㊗️㊗️㊗️㊗️💮💮💮💮💮😻😻😻😻😎😎😎😎🙏🙏🙏💯💯💯💯✨✨✨✨ 125W にpowerlimit でも 24000 examples/sec と, 1080 Ti 135W で 10500 の 2 倍以上の性能ですね, ありがとうございます pic.twitter.com/RqVCSS7GVk
— Syoyo Fujita (@syoyo) July 28, 2018
24000 examples/sec でました.
125W 1080 Ti で 10500 ~ 11500 examples/sec くらいなので, 2 倍以上効率がよいですね!
水冷化しているので熱暴走せずに長期に学習が回せることが期待できますね!
TODO
- 優秀な TensorFlow 若人さまが, 人類史上最速で優秀な水冷 TITAN V で TensorFlow 機械学習若人へと昇華なされるスキームを確立する旅に出たい.
- CUDA 9.1 で安定動作するか試す.
- CUDA 9.0 で動くか試す.