8000 GitHub - ConnorBaker/cuda-packages: A playground for Nix and CUDA
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

ConnorBaker/cuda-packages

Repository files navigation

cuda-packages

Out of tree (Nixpkgs) experiments with packaging CUDA in an extensible way.

Important

I am in the process of upstreaming this into Nixpkgs. This repository will be archived once that is complete.

Most code lives in Nixpkgs and is copied/modified here for ease of development.

Top-level:

  • cudaLib: data and utility functions used in creation of the CUDA package sets
  • cudaPackagesExtensions: extensions (overlays) applied to each CUDA package set
    • an easy way to add packages to all CUDA package sets

Notes

  • Python wrappers which invoke CMake do not always pass their environment to the CMake process. That means a number of the environment variables we set so CMake's auto-detection functionality just works is broken.
  • autoAddDriverRunpath for CMake projects is a crutch -- the correct fix is to have the CMake project link against CUDA::cudart.
  • 12.2.2 is kept around because it is the last version of CUDA 12 supported by Xavier through cuda_compat
    • DO NOT rely on it being around forever -- try to upgrade to newer hardware!
  • cuda_compat can be disabled by setting the package to null. This is useful in cases where the host OS has a recent enough CUDA driver that the compatibility library isn't needed.
  • nvprune can only be run on relocatable libraries, so it wouldn't be helpful as a hook because NVIDIA's dynamic libraries are not relocatable, and we generally don't use the static libraries.

Todo

About

A playground for Nix and CUDA

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

0