Skip to content

Commit e054aea

Browse files
committed
Add H100SXM benchmark
1 parent 8ac984d commit e054aea

File tree

4 files changed

+607
-337
lines changed

4 files changed

+607
-337
lines changed

README.md

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -50,10 +50,12 @@ For details, visit the [preprint](https://www.biorxiv.org/content/biorxiv/early/
5050

5151
Any GPU [supported](https://numba.pydata.org/numba-doc/dev/cuda/overview.html#requirements) by Numba can be used. We tested a number of GPUs:
5252

53-
- RTX 2070, used on a local machine
54-
- T4 GPU, offered for free on Colab
55-
- RTX 4090 GPU, offered on vast.ai
56-
- Any A100 GPU, offered on vast.ai
53+
- GTX 1050 Ti
54+
- RTX 2070
55+
- T4 GPU (free on [Google Colab](https://colab.research.google.com/))
56+
- RTX 4090 (on vast.ai)
57+
- A100 80GB (on vast.ai)
58+
- H100SXM 80GB (on vast.ai)
5759

5860
The `pytorch/pytorch:2.2.1-cuda12.1-cudnn8-devel` docker [image](https://hub.docker.com/layers/pytorch/pytorch/2.2.1-cuda12.1-cudnn8-devel/images/sha256-42204bca460bb77cbd524577618e1723ad474e5d77cc51f94037fffbc2c88c6f?context=explore) was used for development and testing.
5961

0 commit comments

Comments
 (0)