squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Bump CUTLASS Python container version (#672)

Browse Source 
* Update example 40 README

* Update CUTLASS Python README
This commit is contained in:
Jack Kosaian 2022-10-22 21:09:39 -04:00 committed by GitHub
parent 7d0dd6706e
commit 8c1bf9b784
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/40_cutlass_py/README.md
View File

@ -1,9 +1,9 @@
# CUTLASS Python Interface Example # CUTLASS Python Interface Example
## Using Docker ## Using Docker
You can run the PyCUTLASS on NGC pytorch container. You can run the PyCUTLASS on NGC PyTorch container.
```shell ```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 PyCUTLASS requires additional dependency Boost C++ library, which can be installed with
```bash ```bash

4
tools/library/scripts/pycutlass/README.md
View File

@ -81,9 +81,9 @@ The tiling size of above operations can also be customized.
## Installation ## Installation
### Using Docker ### Using Docker
You can run the PyCUTLASS on NGC pytorch container. You can run the PyCUTLASS on NGC PyTorch container.
```shell ```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 PyCUTLASS requires additional dependency Boost C++ library, which can be installed with
```bash ```bash