diff --git a/requirements.txt b/requirements.txt index f4dd2ca..30c70ae 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,2 @@ -pytorch_lightning - +pytorch-lightning cupy-cuda12x