파이토치 설치하기 ROCm 버전 업데이트 (PyTorchKorea#54)

yskelg · Aug 29, 2024 · 4fdd09f · 4fdd09f
1 parent e93b881
commit 4fdd09f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/assets/quick-start-module.js b/assets/quick-start-module.js
@@ -12,8 +12,8 @@ var archInfoMap = new Map([
 ]);
 
 let version_map = {
-  nightly: { accnone: ["cpu", ""], "cuda.x": ["cuda", "11.8"], "cuda.y": ["cuda", "12.1"], "rocm5.x": ["rocm", "6.0"] },
-  release: { accnone: ["cpu", ""], "cuda.x": ["cuda", "11.8"], "cuda.y": ["cuda", "12.1"], "rocm5.x": ["rocm", "5.7"] },
+  nightly: { accnone: ["cpu", ""], "cuda.x": ["cuda", "11.8"], "cuda.y": ["cuda", "12.1"], "rocm5.x": ["rocm", "6.2"] },
+  release: { accnone: ["cpu", ""], "cuda.x": ["cuda", "11.8"], "cuda.y": ["cuda", "12.1"], "rocm5.x": ["rocm", "6.1"] },
 };
 let stable_version = "Stable (2.3.1)";