Jetson Nano에 Tensorflow 설치

예비캡스톤 프로젝트에서 비접촉 키오스크를 만드는데 필요한 코드에서 tensorflow를 사용하기 때문에 처음엔 linux에서 tensorflow를 설치하는 방법을 찾아봤는데 어떤 방법을 써도 설치에 실패했다..

linux 오픈카톡방에서 질문한 뒤 얻어낸 링크-> NVIDIA 공식 사이트에 올라와있는 Jetson Nano용 tensorflow 설치 방법이다.

설명을 읽어보니 tensorflow-1.x 의 경우에는 cpu버전과 gpu버전이 따로 나뉘어 있다. 따라서 gpu를 사용하도록 하려면 tensorflow-1.x-gpu와 같이 특정 gpu전용 버전을 설치해주어야 한다. 

반면에 tensorflow2버전 부터는 gpu가 필요한 작업량이면 알아서 gpu를 사용한다고 한다.

Jetson Nano는 gpu가 내장되어 있기 때문에 알아서 사용할 수 있도록 tensorflow2버전을 깔았다.

만약 내 버전이 tensorflow2버전이고 코드는 1버전으로 작성되어있을 때에는 tensorflow.compat.v1코드를 코드 상단에입력하여 코드를 tensorflow2로 돌릴 수 있도록 변환해주면 된다. 

 

아래 링크로 들어가서 이 코드를 터미널에 작성하면 된다. 

tensorflow2 설치

docs.nvidia.com/deeplearning/frameworks/install-tf-jetson-platform/index.html#overview

Installing TensorFlow For Jetson Platform :: NVIDIA Deep Learning Frameworks Documentation

설치 후 잘 설치가 되었는지 확인하는 방법은 python을 실행시킨 상태에서 import tensorflow 를 실행했을 때 오류없이 다음줄로 넘어가면 잘 설치가 된것이다.