$ sudo apt-get update
$ sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran
$ sudo apt-get install python3-pip
$ sudo python3 -m pip install --upgrade pip
$ sudo pip3 install -U testresources setuptools==65.5.0
$ sudo pip3 install -U numpy==1.22 future==0.18.2 mock==3.0.5 keras_preprocessing==1.1.2 keras_applications==1.0.8 gast==0.4.0 protobuf pybind11 cython pkgconfig packaging h5py==3.7.0
TensorFlowのインストール
$ sudo pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v61 tensorflow==2.16.1+nv24.08
これで、インストールが完了です。
$ python -c "import tensorflow as tf; print( tf.__version__ )" 2> /dev/null
2.16.1
$ pip show tensorflow
Name: tensorflow
Version: 2.16.1+nv24.8
Summary: TensorFlow is an open source machine learning framework for everyone.
Home-page: https://www.tensorflow.org/
Author: Google Inc.
Author-email: packages@tensorflow.org
License: Apache 2.0
Location: /usr/local/lib/python3.10/dist-packages
Requires: absl-py, astunparse, flatbuffers, gast, google-pasta, grpcio, h5py, keras, libclang, ml-dtypes, numpy, opt-einsum, packaging, protobuf, requests, setuptools, six, tensorboard, tensorflow-io-gcs-filesystem, termcolor, typing-extensions, wrapt
Required-by:
$ python -c "from tensorflow.python.client import device_lib; print(device_lib.list_local_devices())" 2> /dev/null
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 13175382782984246198
xla_global_id: -1
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 61077245952
locality {
bus_id: 1
links {
}
}
incarnation: 2764486832775246996
physical_device_desc: "device: 0, name: Orin, pci bus id: 0000:00:00.0, compute capability: 8.7"
xla_global_id: 416903419
]
$ python -m pip install -U protobuf tf_slim tensorflow_datasets keras keras-tuner keras-visualizer scikit-learn
$ python -m pip install git+https://github.com/tensorflow/docs
$ python -m pip install git+https://github.com/tensorflow/examples.git
$ python -m pip install git+https://www.github.com/keras-team/keras-contrib.git
# python -m pip install -U pillow pydot matplotlib seaborn pandas scipy opencv-python opencv-contrib-python