diff --git a/.github/workflows/pytorch.yml b/.github/workflows/pytorch.yml index 1fbbe82a..a0408d75 100644 --- a/.github/workflows/pytorch.yml +++ b/.github/workflows/pytorch.yml @@ -14,7 +14,7 @@ jobs: runs-on: ubuntu-latest strategy: matrix: - python-version: ['3.7', '3.8', '3.9', '3.10'] + python-version: ['3.8', '3.9', '3.10', '3.11'] steps: - uses: actions/checkout@v3 diff --git a/pytorch/requirements.txt b/pytorch/requirements.txt index 9cf29ab8..ddbad7ea 100644 --- a/pytorch/requirements.txt +++ b/pytorch/requirements.txt @@ -1,9 +1,9 @@ mpi4py plotly pytorch-ignite -pytorch-lightning>=1.6.0 +pytorch-lightning skorch -torch==1.11.0 -torchvision==0.12.0 -torchaudio==0.11.0 +torch +torchvision +torchaudio optuna