pinokiod 5.1.24 → 5.1.25

package/kernel/bin/cuda.js

@@ -2,19 +2,7 @@ const semver = require('semver')
 const fs = require('fs')
 const path = require('path')
 class Cuda {
-  async restoreCudaActivationScript() {
-    if (this.kernel.platform !== "win32") {
-      return
-    }
-    const folder = this.kernel.bin.path("miniconda/etc/conda/activate.d")
-    try {
-      await fs.promises.rename(
-        path.resolve(folder, "~cuda-nvcc_activate.bat.disabled"),
-        path.resolve(folder, "~cuda-nvcc_activate.bat")
-      )
-    } catch (e) {}
-  }
-  async stashCompilerVarsScripts() {
+  async stashActivationScripts() {
     if (this.kernel.platform !== "win32") {
       return
     }
@@ -22,6 +10,7 @@ class Cuda {
     const stash = path.resolve(folder, "pinokio")
     await fs.promises.mkdir(stash, { recursive: true }).catch(() => {})
     const scripts = [
+      "~cuda-nvcc_activate.bat",
       "vs2019_compiler_vars.bat",
       "vs2022_compiler_vars.bat",
     ]
@@ -55,8 +44,7 @@ class Cuda {
             "conda install -y nvidia/label/cuda-12.8.1::cuda"
           ]
         }, ondata)
-        await this.restoreCudaActivationScript()
-        await this.stashCompilerVarsScripts()
+        await this.stashActivationScripts()
       } else {
         await this.kernel.bin.exec({
           message: [
@@ -83,8 +71,7 @@ class Cuda {
   async installed() {
     if (this.kernel.gpu === "nvidia") {
       if (this.kernel.platform === 'win32') {
-        await this.restoreCudaActivationScript()
-        await this.stashCompilerVarsScripts()
+        await this.stashActivationScripts()
         if (this.kernel.bin.installed.conda.has("cudnn") && this.kernel.bin.installed.conda.has("cuda") && this.kernel.bin.installed.conda.has("libzlib-wapi")) {
           let version = this.kernel.bin.installed.conda_versions.cuda
           if (version) {

package/kernel/shell.js

@@ -992,6 +992,20 @@ class Shell {
           } else if (vcvars_path && arg) {
             conda_activation.push(`CALL "${vcvars_path}" ${arg} > ${compiler_log} 2>&1`)
           }
+          const cuda_candidates = [
+            path.resolve(activate_root, "pinokio", "~cuda-nvcc_activate.bat"),
+            path.resolve(activate_root, "~cuda-nvcc_activate.bat"),
+          ]
+          let cuda_script = null
+          for (const candidate of cuda_candidates) {
+            if (await this.exists(candidate)) {
+              cuda_script = candidate
+              break
+            }
+          }
+          if (cuda_script) {
+            conda_activation.push(`CALL "${cuda_script}" > ${compiler_log} 2>&1`)
+          }
           conda_activation.push("@echo off")
           conda_activation.push("set TORCH_CUDA_ARCH_LIST=")
           conda_activation.push("set CUDAARCHS=")

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pinokiod",
-  "version": "5.1.24",
+  "version": "5.1.25",
   "description": "",
   "main": "index.js",
   "scripts": {