...
본 문서는 Jetson board에서 Tensorflow를 GPU 사용하는 버전으로 설치하는 과정을 서술합니다.
모든 경우는 초기에 설치 할 경우 작동하며, 다른 python 패키지를 설치한 후에 아래 과정을 따라할 경우 에러가 발생할 수 있습니다.
Jetpack 5.x 에서 설치 과정
Tensorflow 설치
...
| 정보 |
|---|
설치하려는 TensorFlow 버전은 사용 중인 JetPack 버전에서 지원되어야 합니다. 또한 이전 릴리스에서는 패키지 이름이 다를 수 있습니다. 위 예시에서 v502는 사용중인 jetpack version, 2.9.1은 tensorflow version, 22.09는 tensorflow의 월간 NVIDIA 컨테이너 버전을 나타냅니다. Jetson 플랫폼용 TensorFlow 릴리스 노트와 tensorflow download를 참조하여 호환되는 버전으로 다운로드를 진행하시길 바랍니다. |
...
위 과정을 진행한 후 True가 나올경우 GPU 사용
Jetpack 4.x 에서 설치 과정 (Nano)
Tensorflow 설치
apt update
| 코드 블럭 |
|---|
sudo apt-get update |
...
| 코드 블럭 |
|---|
sudo apt-get install -y libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran
sudo ln -s /usr/include/locale.h /usr/include/xlocale.h
sudo pip3 install --verbose 'protobuf<4' 'Cython<3'
sudo pip3 install -U pip testresources setuptools |
tensorflow 다운로드 및 설치 (30분에서 1시간정도 소요됩니다)
| 코드 블럭 |
|---|
sudo wget --no-check-certificate https://developer.download.nvidia.com/compute/redist/jp/v461/tensorflow/tensorflow-2.7.0+nv22.1-cp36-cp36m-linux_aarch64.whl
sudo pip3 install --verbose tensorflow-2.7.0+nv22.1-cp36-cp36m-linux_aarch64.whl
sudo pip3 install numpy==1.19.4 |
Tensorflow GPU 사용 여부 확인
| 코드 블럭 |
|---|
python3 >>> import tensorflow as tf >>> tf.test.is_gpu_available() True |
...
위 과정을 진행한 후 True가 나올경우 GPU 사용
Jetpack 4.x 에서 설치 과정 (TX2)
Tensorflow 설치
apt update
| 코드 블럭 |
|---|
sudo apt-get update |
python3-pip 및 tensorflow에 필요한 패키지 설치
| 코드 블럭 |
|---|
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 pip3 install -U pip testresources setuptools
sudo ln -s /usr/include/locale.h /usr/include/xlocale.h
sudo pip3 install -U numpy==1.19.4 future mock keras_preprocessing keras_applications gast==0.2.1 protobuf pybind11 cython pkgconfig packaging |
tensorflow 다운로드 및 설치
| 코드 블럭 |
|---|
sudo pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v461 tensorflow |
Tensorflow GPU 사용 여부 확인
| 코드 블럭 |
|---|
python3
>>> import tensorflow as tf
>>> tf.config.list_physical_devices('GPU')
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')] |
...
physicalDevice가 empty list가 아니며, 위 내용과 같이 나올 경우 GPU 사용 가능함