文件名称：onnx-keras-example
文件大小：5KB
文件格式：ZIP
更新时间：2024-04-14 05:27:39
Python
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.训练模型并测试
【文件预览】：
onnx-keras-example-main
----Dockerfile.CUDA(231B)
----training()
--------train.py(2KB)
----README.md(2KB)
----Dockerfile.CPU(154B)
----requirements.CPU.txt(73B)
----requirements.CUDA.txt(77B)
----prediction()
--------predict_default.py(945B)
--------predict_onnx.py(1KB)