...
위 과정을 진행한 후 True가 나올경우 GPU 사용
Jetpack 4.x 에서 설치 과정 (Nano)
Tensorflow 설치
apt update
코드 블럭 |
---|
sudo apt-get update |
...
위 과정을 진행한 후 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 사용 가능함