CUDA: Difference between revisions

← Older edit Newer edit →

VisualWikitext

@@ Line 20: / Line 20: @@
 {{file|cuda-fhs.nix|nix|<nowiki>
-{ pkgs ? import <nixpkgs> {} }:
+# Run with `nix-shell cuda-fhs.nix`
+{ pkgs ? import </nowiki><nixpkgs><nowiki> {} }:
-let
+(pkgs.buildFHSUserEnv {
-  fhs = pkgs.buildFHSUserEnv {
+  name = "cuda-env";
-    name = "cuda-env";
+  targetPkgs = pkgs: with pkgs; [
-    targetPkgs = pkgs: with pkgs; [
+    git
-      git
+    gitRepo
-      gitRepo
+    gnupg
-      gnupg
+    autoconf
-      autoconf
+    curl
-      curl
+    procps
-      procps
+    gnumake
-      gnumake
+    util-linux
-      util-linux
+    m4
-      m4
+    gperf
-      gperf
+    unzip
-      unzip
+    cudatoolkit
-      cudatoolkit
+    linuxPackages.nvidia_x11
-      linuxPackages.nvidia_x11
+    libGLU libGL
-      libGLU libGL
+    xorg.libXi xorg.libXmu freeglut
-      xorg.libXi xorg.libXmu freeglut
+    xorg.libXext xorg.libX11 xorg.libXv xorg.libXrandr zlib
-      xorg.libXext xorg.libX11 xorg.libXv xorg.libXrandr zlib
+    ncurses5
-      ncurses5
+    stdenv.cc
-      stdenv.cc
+    binutils
-      binutils
+  ];
-    ];
+  multiPkgs = pkgs: with pkgs; [ zlib ];
-    multiPkgs = pkgs: with pkgs; [ zlib ];
+  runScript = "bash";
-    runScript = "bash";
+  profile = ''
-    profile = ''
+    export CUDA_PATH=${pkgs.cudatoolkit}
-       export CUDA_PATH=${pkgs.cudatoolkit}
+    # export LD_LIBRARY_PATH=${pkgs.linuxPackages.nvidia_x11}/lib
-       # export LD_LIBRARY_PATH=${pkgs.linuxPackages.nvidia_x11}/lib
+    export EXTRA_LDFLAGS="-L/lib -L${pkgs.linuxPackages.nvidia_x11}/lib"
-       export EXTRA_LDFLAGS="-L/lib -L${pkgs.linuxPackages.nvidia_x11}/lib"
+    export EXTRA_CCFLAGS="-I/usr/include"
-       export EXTRA_CCFLAGS="-I/usr/include"
+  '';
-     '';
+}).env
-  };
-in pkgs.stdenv.mkDerivation {
-   name = "cuda-env-shell";
-   nativeBuildInputs = [ fhs ];
-   shellHook = "exec cuda-env";
-}
 </nowiki>}}
@@ Line 65: / Line 59: @@
 * By making a nix-shell
 {{file|cuda-shell.nix|nix|<nowiki>
-{ pkgs ? import <nixpkgs> {} }:
+# Run with `nix-shell cuda-shell.nix`
+{ pkgs ? import </nowiki><nixpkgs><nowiki> {} }:
-pkgs.stdenv.mkDerivation {
+pkgs.mkShell {
     name = "cuda-env-shell";
     buildInputs = with pkgs; [