mirror of
https://github.com/nod-ai/AMD-SHARK-Studio.git
synced 2026-02-19 11:56:43 -05:00
0ee515a7be
* Add CUDA_BENCHMARKS option to setup.venv to enable PyTorch benchmarks on CUDA. * Fix PyTorch GPU benchmarks for tank models.
14 lines
103 B
Plaintext
14 lines
103 B
Plaintext
setuptools
|
|
wheel
|
|
|
|
# SHARK Runner
|
|
tqdm
|
|
|
|
# SHARK Downloader
|
|
gsutil
|
|
|
|
# Testing
|
|
pytest
|
|
pytest-xdist
|
|
Pillow
|