diff --git a/requirements.txt b/requirements.txt index c83c1f6..f4dd2ca 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,3 @@ pytorch_lightning -cupy-cuda11x \ No newline at end of file + +cupy-cuda12x