yindo/flash-attention-prebuild-wheels-rocm
1
0
Fork 0
mirror of https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git synced 2026-07-01 01:37:53 -04:00
Code Issues Packages Projects Releases 2 Wiki Activity

torch v2.0.0 not supported cuda 12.6

Browse Source
This commit is contained in:
Junya Morioka
2025-02-04 17:07:01 +09:00
parent d1b13bc000
commit 74e18e7877
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build.yml
+2
View File
@@ -50,6 +50,8 @@ jobs:
cuda-version: "12.1.1"
- torch-version: "2.0.1"
cuda-version: "12.4.1"
- torch-version: "2.0.1"
cuda-version: "12.6.3"
steps:
- uses: actions/checkout@v4