From 19edb4f8a21619f30e263a11ff4f36d02dda6c5a Mon Sep 17 00:00:00 2001 From: Red Date: Mon, 17 Mar 2025 12:53:32 -0400 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index eaff9c4a..7007757c 100644 --- a/README.md +++ b/README.md @@ -271,7 +271,7 @@ For models compatible with Cambricon Extension for PyTorch (torch_mlu). Here's a # DOCKER -This uses the base image nvidia/cuda:12.6.3-cudnn-runtime-ubuntu24.04 and stable torch version from https://download.pytorch.org/whl/cu126. +FOR Nvidia and AMD supported by ROCm After cloning the repo, run the following command once in bash terminal to build the docker image: