Commit ·
30922b0
1
Parent(s): e4c31d4
fix cuda
Browse files
update-k2-doc-cuda-linux.py
CHANGED
|
@@ -109,6 +109,7 @@ def remove_all_files(d: str):
|
|
| 109 |
|
| 110 |
def get_all_cuda_wheels():
|
| 111 |
cuda = get_all_files("cuda", suffix="*.whl")
|
|
|
|
| 112 |
cuda_wheels = generate_url(cuda)
|
| 113 |
return cuda_wheels
|
| 114 |
|
|
|
|
| 109 |
|
| 110 |
def get_all_cuda_wheels():
|
| 111 |
cuda = get_all_files("cuda", suffix="*.whl")
|
| 112 |
+
cuda += get_all_files("ubuntu-cuda", suffix="*.whl")
|
| 113 |
cuda_wheels = generate_url(cuda)
|
| 114 |
return cuda_wheels
|
| 115 |
|