NX入门教程软件篇-安装keras

NX入门教程软件篇-安装keras

说明：

• 介绍如何安装keras
• pytyon3 + keras 2.4.3

步骤：

• 前提是要先TensorFlow

# beforehand, install TensorFlow (https://eLinux.org/Jetson_Zoo#TensorFlow)
$ sudo apt-get install -y build-essential libatlas-base-dev gfortran
$ pip3 install keras

测试：

• 新建文件 vim hello_world.py
• 内容如下：

import tensorflow as tf
from tensorflow.keras import layers
print(tf.__version__)
print(tf.keras.__version__)

• 效果如下：

$ python3  hello_world.py 
2021-07-02 20:49:24.027922: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.10.2
2.3.1
2.4.0