...
본 문서는 Jetson board에서 Tensorflow를 GPU 사용하는 버전으로 설치하는 과정을 서술합니다.
모든 경우는 초기에 설치 할 경우 작동하며, 다른 python 패키지를 설치한 후에 아래 과정을 따라할 경우 에러가 발생할 수 있습니다.
Jetpack 5.x 에서 설치 과정
Tensorflow 설치
...
| 코드 블럭 |
|---|
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 |
...
physicalDevice가 empty list가 아니고아니며, 위 내용과 같이 나올 경우 GPU 사용 가능함