From fbe9283c9d705660e871c7cc59b6b6be0e58501b Mon Sep 17 00:00:00 2001
From: Jalal El Mansouri <84198762+jelmansouri-legion@users.noreply.github.com>
Date: Sun, 19 Sep 2021 10:17:57 -0400
Subject: [PATCH] cuda: Fix installation script (#2406)

* Fix cuda installation script

* update to version 11.4.2_471.41

Co-authored-by: Issac Lin <issaclin32@gmail.com>
---
 bucket/cuda.json | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/bucket/cuda.json b/bucket/cuda.json
index 992ffec7a4..fb0b115986 100644
--- a/bucket/cuda.json
+++ b/bucket/cuda.json
@@ -14,17 +14,15 @@
     },
     "installer": {
         "script": [
-            "$names = @('bin', 'extras', 'include', 'jre', 'lib', 'libnvvp', 'nvml', 'nvvm')",
+            "$names = @('bin', 'extras', 'include', 'lib', 'libnvvp', 'nvml', 'nvvm')",
             "foreach ($name in $names) {",
-            "    Copy-Item \"$dir\\*\\$name\" \"$dir\" -Recurse -Force",
+            "    Copy-Item \"$dir\\cuda_*\\*\\$name\" \"$dir\" -Recurse -Force",
+            "    Copy-Item \"$dir\\lib*\\*\\$name\" \"$dir\" -Recurse -Force",
             "}",
             "Get-ChildItem \"$dir\" -Exclude $names | Remove-Item -Recurse -Force"
         ]
     },
-    "env_add_path": [
-        "bin",
-        "libnvvp"
-    ],
+    "env_add_path": "bin",
     "env_set": {
         "CUDA_PATH": "$dir"
     },