publishing

Oneflow-Inc · Dec 6, 2023 · fbd0c3c · fbd0c3c
1 parent d388c95
commit fbd0c3c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/dist/index.js b/dist/index.js
diff --git a/src/utils/docker.ts b/src/utils/docker.ts
@@ -172,12 +172,14 @@ async function buildAndMakeWheel(
   let nvLibs: string[] = []
   if (useNVWheels) {
     nvLibs = [
-      'libcudnn_cnn_infer.so',
-      'libcudnn_cnn_train.so',
-      'libcudnn_ops_infer.so',
-      'libcudnn_ops_train.so',
-      'libcublas.so',
-      'libcublasLt.so'
+      'libcudnn_cnn_infer.so.8',
+      'libcudnn_cnn_train.so.8',
+      'libcudnn_ops_infer.so.8',
+      'libcudnn_ops_train.so.8',
+      'libcublas.so.11',
+      'libcublasLt.so.11',
+      'libcublas.so.12',
+      'libcublasLt.so.12'
     ]
   }
   const nvLibsExcludes = Array.prototype.concat.apply(