こんにちは．今回は 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 でも使えます！

参考

Accelerating TensorFlow on NVIDIA A100 GPUs | NVIDIA Technical Blog

The NVIDIA A100, based on the NVIDIA Ampere GPU architecture, offers a suite of exciting new features: third-generation Tensor Cores, Multi-Instance GPU (MIG) and third-generation NVLink.

https://developer.nvidia.com/blog/accelerating-tensorflow-on-a100-gpus/

RTX3070などの30xx系GPU かつ CUDA11.3 インストール済みのホストPC上でTensorFlow v1.15.5を動かす

https://zenn.dev/pinto0309/articles/9e54ee8d15189a

GitHub - NVIDIA/tensorflow: An Open Source Machine Learning Framework for Everyone

An Open Source Machine Learning Framework for Everyone - GitHub - NVIDIA/tensorflow: An Open Source Machine Learning Framework for Everyone

https://github.com/NVIDIA/tensorflow