Tensorflowのインストール

PC環境

• OS: Ubuntu 20.04 LTS

• GPU: NVIDIA GeForce RTX2070 SUPER

• NVIDIAドライバ: 535.54.03

• CUDAバージョン: 12.2

• cuDNNバージョン: 8.9.2.26

• Python: 3.8.10

• Tensorflow: 2.11.0

GPUセットアップはこちらの記事を参照：

仮想環境セットアップ

$ python3 --version
Python 3.8.10

$ sudo apt update
$ sudo apt install python3.8-venv
$ python3 -m venv .venv_tensorflow-2.11
$ source .venv_tensorflow-2.11/bin/activate

$ pip install -U pip
$ pip --version
pip 23.1.2 from /home/XXX/.venv_tensorflow-2.11/lib/python3.8/site-packages/pip (python 3.8)

Tensorflowをインストール

$ pip install tensorflow==2.11.0

インストール確認

$ python
>>> import tensorflow as tf
>>> print(tf.reduce_sum(tf.random.normal([1000, 1000])))
...
tf.Tensor(181.74527, shape=(), dtype=float32)
>>> print(tf.config.list_physical_devices('GPU'))
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]

1つ目のprint文でTensorが返ってくれば、TensorFlowは正しくインストールされている。
2つ目のprint文でGPUデバイスのリストが返ってくれば、GPUもOK.
思ったよりもスムーズにインストールできた。

---

参考URL:

Install TensorFlow with pip