amdih/ryzen-ai-pytorch repository overview

⁠Ryzen AI Pytorch

The Ryzen AI Pytorch Docker image provides the quantizer tool to quantize pretrained Pytorch 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-pytorch: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-pytorch:latest

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