Update test-gpu.sh

yfukai · Mar 12, 2024 · 76c3b68 · 76c3b68
1 parent 6dd7e0e
commit 76c3b68
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test-gpu.sh b/test-gpu.sh
@@ -3,4 +3,4 @@ version="0.3.1"
 docker pull ghcr.io/yfukai/basicpy-docker-mcmicro:${version}-cuda
 docker run --rm -v $(pwd)/testdata:/data ghcr.io/yfukai/basicpy-docker-mcmicro:${version}-cuda -d=gpu -i /data/exemplar-001-cycle-06.ome.tiff -o /data/
 docker pull ghcr.io/yfukai/basicpy-docker-mcmicro:latest-cuda
-docker run --rm -v $(pwd)/testdata:/data ghcr.io/yfukai/basicpy-docker-mcmicro:latest -d=gpu -i /data/exemplar-001-cycle-06.ome.tiff -o /data/
+docker run --rm -v $(pwd)/testdata:/data ghcr.io/yfukai/basicpy-docker-mcmicro:latest-cuda -d=gpu -i /data/exemplar-001-cycle-06.ome.tiff -o /data/