TrueNAS/scale-build
1
0
Fork 0
mirror of https://github.com/truenas/scale-build.git synced 2026-02-15 07:29:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use wget to download NVidia driver (#811)

Browse Source
This commit is contained in:
themylogin
2025-01-23 20:49:00 +01:00
committed by GitHub
parent 1736d42338
commit 4d4b5a08bc
1 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scale_build/extensions.py
View File

@@ -120,7 +120,6 @@ class NvidiaExtension(Extension):
binaries = ("apt", "apt-config", "apt-key", "dpkg")
temporary_packages = ["gcc", "make", "pkg-config"]
permanent_packages = ["libvulkan1", "nvidia-container-toolkit", "vulkan-validationlayers"]
headers = {"User-Agent": "curl/7.88.1"}
def build_impl(self):
kernel_version = get_kernel_version(self.chroot)
@@ -158,10 +157,7 @@ class NvidiaExtension(Extension):
filename = f"NVIDIA-Linux-x86_64-{version}-no-compat32.run"
result = f"{self.chroot}/{filename}"
with requests.get(f"{prefix}/{version}/{filename}", headers=self.headers, stream=True, timeout=10) as r:
r.raise_for_status()
with open(result, "wb") as f:
shutil.copyfileobj(r.raw, f)
self.run(["wget", "-c", "-O", f"/{filename}", f"{prefix}/{version}/{filename}"])
os.chmod(result, 0o755)
return result