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