diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml index 5538f8aa..fa3e8bb3 100644 --- a/.github/workflows/docker-build.yml +++ b/.github/workflows/docker-build.yml @@ -28,7 +28,6 @@ jobs: python: - "3.10" pytorch: - - "2.1.1" - "2.1.2" - "2.2.0" steps: @@ -100,7 +99,6 @@ jobs: python: - "3.10" pytorch: - - "2.1.1" - "2.1.2" - "2.2.0" cuda: @@ -177,12 +175,10 @@ jobs: python: - "3.10" pytorch: - - "2.1.1" - "2.1.2" - "2.2.0" rocm: - "5.7" - - "5.7" level: - "runtime" exclude: @@ -190,8 +186,6 @@ jobs: pytorch: "2.2.0" - rocm: "5.7" pytorch: "2.1.2" - - rocm: "5.7" - pytorch: "2.1.1" steps: - diff --git a/README.md b/README.md index 157cedc5..6675690a 100644 --- a/README.md +++ b/README.md @@ -88,7 +88,7 @@ You can also [build from source](#building-images) by editing `.env` and running Supported Python versions: `3.11`, `3.10` -Supported Pytorch versions: `2.2.0`, `2.1.2`, `2.1.1` +Supported Pytorch versions: `2.2.0`, `2.1.2` Supported Platforms: `NVIDIA CUDA`, `AMD ROCm`, `CPU`