feat(nvidia): CUDA toolkit

atomic-studio-org · Mar 8, 2024 · 8f0675d · 8f0675d
1 parent 46c24ac
commit 8f0675d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/config/common/shared/nvidia/scripts.yml b/config/common/shared/nvidia/scripts.yml
@@ -1,3 +1,4 @@
 type: script-nu
 scripts:
   - nvidia-explicit-sync.nu
+  - cuda-toolkit.nu
diff --git a/config/scripts/cuda-toolkit.nu b/config/scripts/cuda-toolkit.nu
@@ -0,0 +1,12 @@
+#!/usr/bin/env -S nu 
+
+use lib/std.nu [ARCH]
+
+let FEDORA_MAJOR_VERSION = (run-external --redirect-combine rpm '-E' '%fedora' | complete).stdout
+const COPR_FILE = "/etc/yum.repos.d/_cuda-toolkit.repo"
+
+http get $"https://developer.download.nvidia.com/compute/cuda/repos/fedora($FEDORA_MAJOR_VERSION)/($ARCH)/cuda-fedora($FEDORA_MAJOR_VERSION).repo" | save -f $COPR_FILE 
+
+rpm-ostree install cuda-toolkit-12-4
+
+rm -f $COPR_FILE