diff options
Diffstat (limited to 'pyproject.toml')
| -rw-r--r-- | pyproject.toml | 20 |
1 files changed, 16 insertions, 4 deletions
diff --git a/pyproject.toml b/pyproject.toml index e444258..b535304 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -14,13 +14,25 @@ python = "^3.9" loguru = "^0.6.0" omegaconf = "^2.1.0" einops = "^0.3.0" -pytorch-lightning = "^1.7.6" hydra-core = "^1.2.0" smart-open = "^5.2.1" -torch = "^1.11.0" -torchvision = "^0.12.0" +torch = ">=2.0.0" +torchvision = "^0.15.0" adan-pytorch = "^0.1.0" torchmetrics = "^0.9.3" +pytorch-lightning = "^2.0.7" +nvidia-cublas-cu11 = "^11.11.3.6" +nvidia-cuda-cupti-cu11 = "^11.8.87" +nvidia-cuda-nvrtc-cu11 = "^11.8.89" +nvidia-cuda-runtime-cu11 = "^11.8.89" +nvidia-cudnn-cu11 = "^8.9.4.25" +nvidia-cufft-cu11 = "^10.9.0.58" +nvidia-curand-cu11 = "^10.3.0.86" +nvidia-cusolver-cu11 = "^11.4.1.48" +nvidia-cusparse-cu11 = "^11.7.5.86" +nvidia-nccl-cu11 = "^2.18.3" +nvidia-nvtx-cu11 = "^11.8.86" +triton = "^2.0.0.post1" [tool.poetry.group.dev.dependencies] pytest = "^5.4.2" @@ -31,7 +43,7 @@ flake8 = "^5.0.4" safety = "^1.9.0" mypy = "^0.770" typeguard = "^2.7.1" -wandb = "0.12.16" +wandb = "0.15.8" scipy = "^1.6.1" jupyterlab = "^3.0.15" ipywidgets = "^7.6.3" |