Skip to content

Commit

Permalink
Merge branch 'dev' of https://github.com/mlcommons/cm4abtf into dev
Browse files Browse the repository at this point in the history
  • Loading branch information
gfursin committed Apr 10, 2024
2 parents 2eeb1e4 + 7587490 commit 96b2de6
Show file tree
Hide file tree
Showing 2 changed files with 25 additions and 3 deletions.
16 changes: 13 additions & 3 deletions README-cuda.md
Original file line number Diff line number Diff line change
@@ -1,14 +1,24 @@
[ [Back to the main page](README.md) ]

*Tested with PyTorch 2.2.2 and CUDA 11.8 and 12.1*

## Prepare workflow to benchmark ABTF model on CUDA-based device

We expect that you already have CUDA driver installed
### Prerequisites

* We expect that you already have CUDA driver installed
* Tested with PyTorch 2.2.2 and CUDA 11.8 and 12.1


### Build MLPerf loadgen

```bash
cmr "get mlperf inference loadgen _copy" --version=main
```


### Detect or install CUDA toolkit and libraries

``bash
```bash
cmr "get cuda _toolkit _cudnn"
cmr "get cuda-devices"
```
Expand Down
12 changes: 12 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -46,6 +46,18 @@ cm pull repo mlcommons@ck --checkout=dev
cm pull repo mlcommons@cm4abtf --checkout=dev
```

### Find repositories

```bash
cm find repo
```

### Update all repositories at any time

```bash
cm pull repo
```


## Download test ABTF model and image

Expand Down

0 comments on commit 96b2de6

Please sign in to comment.