diff --git a/requirements.txt b/requirements.txt index d4594df39..13a46688f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -27,3 +27,10 @@ kornia>=0.7.1 spandrel pydantic~=2.0 pydantic-settings~=2.0 + +# for GDS +cupy-cuda12x>=12.0.0 +pynvml>=11.4.1 +cudf>=23.0.0 +numba>=0.57.0 +nvidia-ml-py>=12.0.0 \ No newline at end of file