publishing

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

Hide file tree

Showing 2 changed files with 7 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,12 @@ async function buildAndMakeWheel(
   let nvLibs: string[] = []
   if (useNVWheels) {
     nvLibs = [
-      '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'
+      'libcudnn_cnn_infer.so',
+      'libcudnn_cnn_train.so',
+      'libcudnn_ops_infer.so',
+      'libcudnn_ops_train.so',
+      'libcublas.so',
+      'libcublasLt.so'
     ]
   }
   const nvLibsExcludes = Array.prototype.concat.apply(