mirror of
https://github.com/nod-ai/AMD-SHARK-Studio.git
synced 2026-04-03 03:00:17 -04:00
* 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
|