onnx keras example 源码

ONNX运行时示例:具有Keras / Tensorflow的简单神经网络 本示例是如何使用ONNX运行时的完整示例。 让我们从进入keras-tf-nn目录开始并遵循以下步骤。 1.构建Docker镜像并运行容器: 要运行演示程序,构建映像,然后启动容器: docker build -t onnx-keras-nn -f Dockerfile.CUDA . docker run -it --gpus=all onnx-keras-nn:latest bash 如果您的设备不支持CUDA, Dockerfile.CPU在build命令Dockerfile.CPU替换为Dockerfile.CUDA , Dockerfile.CPU从run命令中删除--gpus=all 。 如果您的设备支持CUDA,请确保已安装nvidia-container-runtime 。 2.训练模型并测试