From 7a2c0e36808a2cf63366521fb4cecf3573a4c633 Mon Sep 17 00:00:00 2001 From: Josh Bailey Date: Fri, 21 Jun 2024 15:55:47 +1200 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 98cf03c..e3fc293 100644 --- a/README.md +++ b/README.md @@ -18,7 +18,7 @@ https://github.com/pytorch/serve/issues/3176 ```docker run -v $(pwd)/model_store:/model_store -p 8080:8080 --rm --name torchserve -d iqtlabs/torchserve --models =.mar``` -* [iqtlabs/cuda-torchserve](https://hub.docker.com/r/iqtlabs/cuda-torchserve): CUDA (12.4 or later) accelerated for amd64 only. +* [iqtlabs/cuda-torchserve](https://hub.docker.com/r/iqtlabs/cuda-torchserve): CUDA (12.5 or later) accelerated for amd64 only. ```docker run --gpus all -v $(pwd)/model_store:/model_store -p 8080:8080 --rm --name torchserve -d iqtlabs/cuda-torchserve --models =.mar```