From 8c1bf9b784a7878b20fef673892f0a1d769fd94c Mon Sep 17 00:00:00 2001 From: Jack Kosaian Date: Sat, 22 Oct 2022 21:09:39 -0400 Subject: [PATCH] Bump CUTLASS Python container version (#672) * Update example 40 README * Update CUTLASS Python README --- examples/40_cutlass_py/README.md | 4 ++-- tools/library/scripts/pycutlass/README.md | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/examples/40_cutlass_py/README.md b/examples/40_cutlass_py/README.md index dcbcee2a..0c4b4f1e 100644 --- a/examples/40_cutlass_py/README.md +++ b/examples/40_cutlass_py/README.md @@ -1,9 +1,9 @@ # CUTLASS Python Interface Example ## Using Docker -You can run the PyCUTLASS on NGC pytorch container. +You can run the PyCUTLASS on NGC PyTorch container. ```shell -docker run --gpus all -it --rm nvcr.io/nvidia/pytorch:22.08-py3 +docker run --gpus all -it --rm nvcr.io/nvidia/pytorch:22.09-py3 ``` PyCUTLASS requires additional dependency Boost C++ library, which can be installed with ```bash diff --git a/tools/library/scripts/pycutlass/README.md b/tools/library/scripts/pycutlass/README.md index 7b079d36..498e1aa8 100644 --- a/tools/library/scripts/pycutlass/README.md +++ b/tools/library/scripts/pycutlass/README.md @@ -81,9 +81,9 @@ The tiling size of above operations can also be customized. ## Installation ### Using Docker -You can run the PyCUTLASS on NGC pytorch container. +You can run the PyCUTLASS on NGC PyTorch container. ```shell -docker run --gpus all -it --rm nvcr.io/nvidia/pytorch:22.08-py3 +docker run --gpus all -it --rm nvcr.io/nvidia/pytorch:22.09-py3 ``` PyCUTLASS requires additional dependency Boost C++ library, which can be installed with ```bash