pinokiod 1.0.15 → 1.0.16
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/kernel/bin/cuda.js +10 -6
- package/package.json +1 -1
package/kernel/bin/cuda.js
CHANGED
|
@@ -2,18 +2,21 @@ class Cuda {
|
|
|
2
2
|
async install(req, ondata) {
|
|
3
3
|
if (this.kernel.platform === "win32") {
|
|
4
4
|
await this.kernel.bin.exec({
|
|
5
|
-
message: "conda install -y cudnn libzlib-wapi -c conda-forge"
|
|
6
|
-
|
|
5
|
+
message: "conda install -y cudnn libzlib-wapi -c conda-forge",
|
|
6
|
+
conda: "base",
|
|
7
7
|
}, ondata)
|
|
8
8
|
await this.kernel.bin.exec({
|
|
9
|
-
message: "conda install -y cuda -c nvidia/label/cuda-11.8.0"
|
|
9
|
+
message: "conda install -y cuda -c nvidia/label/cuda-11.8.0",
|
|
10
|
+
conda: "base",
|
|
10
11
|
}, ondata)
|
|
11
12
|
} else {
|
|
12
13
|
await this.kernel.bin.exec({
|
|
13
|
-
message: "conda install -y cudnn -c conda-forge"
|
|
14
|
+
message: "conda install -y cudnn -c conda-forge",
|
|
15
|
+
conda: "base",
|
|
14
16
|
}, ondata)
|
|
15
17
|
await this.kernel.bin.exec({
|
|
16
|
-
message: "conda install -y cuda -c nvidia/label/cuda-11.8.0"
|
|
18
|
+
message: "conda install -y cuda -c nvidia/label/cuda-11.8.0",
|
|
19
|
+
conda: "base",
|
|
17
20
|
}, ondata)
|
|
18
21
|
}
|
|
19
22
|
}
|
|
@@ -38,7 +41,8 @@ class Cuda {
|
|
|
38
41
|
// }
|
|
39
42
|
async uninstall(req, ondata) {
|
|
40
43
|
await this.kernel.bin.exec({
|
|
41
|
-
message: "conda remove cudnn cuda"
|
|
44
|
+
message: "conda remove cudnn cuda",
|
|
45
|
+
conda: "base",
|
|
42
46
|
}, ondata)
|
|
43
47
|
}
|
|
44
48
|
}
|