Skip to content

Instantly share code, notes, and snippets.

@cleverca22
Created September 22, 2019 09:14
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save cleverca22/f6ee9044aa4528c3bb52a0b195e9d938 to your computer and use it in GitHub Desktop.
Save cleverca22/f6ee9044aa4528c3bb52a0b195e9d938 to your computer and use it in GitHub Desktop.
with import <nixpkgs> { config.allowUnfree = true; };
stdenv.mkDerivation {
name = "myCUDA";
buildInputs = [ (python37.withPackages (pkgs: [pkgs.tensorflowWithCuda])) ];
LD_LIBRARY_PATH="${linuxPackages.nvidia_x11}/lib";
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment