基本的には、TensorFlowのテスト済みビルド構成表に従ってインストールします。
https://www.tensorflow.org/install/source?hl=ja#gpu
細かいバージョン違いで動かなかったりするので、自分の環境の備忘録。
| Item | Version |
|---|---|
| Ubuntu20.04 | 5.4.0-107-generic #121-Ubuntu |
| CPU | Intel Core i5-9300H |
| GPU | NVIDIA GeForce GTX 1650 Mobile |
| gcc | 9.4.0 |
| nvidia-driver | 460.27.04 |
| cuda | 11.2.0 |
| libcudnn8 | 8.1.0.77+cuda11.2 |
| libcudnn8-dev | 8.1.0.77+cuda11.2 |
| nvidia-machine-learning-repo | ubuntu2004_1.0.0-1 |
| libnvinfer8 | 8.0.0-1+cuda11.3 |
| Python | 3.8.7 |
| tensorflow | 2.6.0 |
| keras | 2.6.0 |