amdih/ryzen-ai-tensorflow2 repository overview

⁠Ryzen AI Tensorflow2

The Ryzen AI TensorFlow2 Docker image provides the quantizer tool to quantize pretrained Tensorflow 2.x models on CPU. A Dockerfile⁠ is provided to build the GPU docker for nVidia GPU.

Follow these steps:

1. Obtain the latest Docker image. docker pull amdih/ryzen-ai-tensorflow2:latest

2. Start a Docker container using the image. docker run -it --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --group-add video --ipc=host --shm-size 8G amdih/ryzen-ai-tensorflow2:latest

   You can also pass the -v argument to mount any data directories from the host onto the container.