こんにちは．今回は RTX3090 や 3080 などの GPU で Tensorflow の 1.15 と Tensorboard を利用する方法をメモします．
公式が配布している Tensorflow で 3000 番台の GPU を利用するには 2 系に上げなければいけません.
Tensorflow は 1 系と 2 系で API の仕様が変更されているため，1 系のコードを動かすのは大変です．
そこで，公式が配布しているものではなく，NVIDIA が配布しているものを使います．
pip install nvidia-pyindex
pip install nvidia-tensorflow
pip uninstall tensorboard
pip install nvidia-tensorboard要件として，
- Ubuntu 20.04 or later
- Nvidia Driver r455
- Python 3.8
- pip 19.0 or later
が求められます．
これで Tensorflow==1.15 & Tensorboard が RTX 3090 でも使えます！
参考
https://developer.nvidia.com/blog/accelerating-tensorflow-on-a100-gpus/