Merge pull request #618 from caponetto/RHOAIENG-9741

RHOAIENG-9741: Add Tensorflow/PyTorch version compatibility files
opendatahub-io · Jul 17, 2024 · 821a004 · 821a004
2 parents e506603 + 615ad96
commit 821a004
Show file tree

Hide file tree

Showing 2 changed files with 780 additions and 0 deletions.
diff --git a/ci/version-compatibility/pytorch.json b/ci/version-compatibility/pytorch.json
@@ -0,0 +1,184 @@
+{
+  "2.3.1": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.18.1",
+    "torchaudio": "2.3.0",
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.7.0.84",
+      "rocm": "6.0"
+    }
+  },
+  "2.3.0": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.18.0",
+    "torchaudio": "2.3.0",
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.7.0.84",
+      "rocm": "6.0"
+    }
+  },
+  "2.2.2": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.17.2",
+    "torchaudio": "2.2.2",
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.7.0.84",
+      "rocm": "5.7"
+    }
+  },
+  "2.2.1": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.17.1",
+    "torchaudio": "2.2.1",
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.7.0.84",
+      "rocm": "5.7"
+    }
+  },
+  "2.2.0": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.17.0",
+    "torchaudio": "2.2.0",
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.7.0.84",
+      "rocm": "5.7"
+    }
+  },
+  "2.1.2": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.16.2",
+    "torchaudio": "2.1.2",
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.7.0.84",
+      "rocm": "5.6"
+    }
+  },
+  "2.1.1": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.16.1",
+    "torchaudio": "2.1.1",
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.7.0.84",
+      "rocm": "5.6"
+    }
+  },
+  "2.1.0": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.16.0",
+    "torchaudio": "2.1.0",
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.7.0.84",
+      "rocm": "5.6"
+    }
+  },
+  "2.0.1": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.15.2",
+    "torchaudio": "2.0.2",
+    "gpu": {
+      "cuda": "11.7",
+      "cudnn": "8.5.0.96",
+      "rocm": "5.4"
+    }
+  },
+  "2.0.0": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.15.0",
+    "torchaudio": "2.0.0",
+    "gpu": {
+      "cuda": "11.7",
+      "cudnn": "8.5.0.96",
+      "rocm": "5.4"
+    }
+  },
+  "1.13.1": {
+    "python": {
+      "min": "3.7",
+      "max": "3.10"
+    },
+    "torchvision": "0.14.1",
+    "torchaudio": "0.13.1",
+    "gpu": {
+      "cuda": "11.6",
+      "cudnn": "8.3.2.44",
+      "rocm": "5.2"
+    }
+  },
+  "1.13.0": {
+    "python": {
+      "min": "3.7",
+      "max": "3.10"
+    },
+    "torchvision": "0.14.0",
+    "torchaudio": "0.13.0",
+    "gpu": {
+      "cuda": "11.6",
+      "cudnn": "8.3.2.44",
+      "rocm": "5.2"
+    }
+  },
+  "1.12.1": {
+    "python": {
+      "min": "3.7",
+      "max": "3.10"
+    },
+    "torchvision": "0.13.1",
+    "torchaudio": "0.12.1",
+    "gpu": {
+      "cuda": "11.3",
+      "cudnn": "8.3.2.44",
+      "rocm": "5.0"
+    }
+  },
+  "1.12.0": {
+    "python": {
+      "min": "3.7",
+      "max": "3.10"
+    },
+    "torchvision": "0.13.0",
+    "torchaudio": "0.12.0",
+    "gpu": {
+      "cuda": "11.3",
+      "cudnn": "8.3.2.44",
+      "rocm": "5.0"
+    }
+  }
+}