|
|
| Line 23: |
Line 23: |
|
| |
|
| = CUDA and using your GPU for compute = | | = CUDA and using your GPU for compute = |
| NixOS supports using NVIDIA GPUs for pure computing purposes, not just for graphics. For example, many users rely on NixOS for machine learning both locally and on cloud instances. These use cases are supported by the [https://github.com/orgs/NixOS/teams/cuda-maintainers @NixOS/cuda-maintainers team] on GitHub. If you have an issue using your NVIDIA GPU for computing purposes [https://github.com/nixos/nixpkgs/issues/new open an issue] on GitHub and tag @NixOS/cuda-maintainers.
| | See the [[CUDA]] wiki page! |
| | |
| Note that you may need to adjust your driver version to use "data center" GPUs like V100/A100s. See [https://discourse.nixos.org/t/how-to-use-nvidia-v100-a100-gpus/17754 this thread] for more info.
| |
| | |
| == CUDA ==
| |
| | |
| The CUDA toolkit is available in a [https://search.nixos.org/packages?channel=unstable&from=0&size=50&buckets=%7B%22package_attr_set%22%3A%5B%22cudaPackages%22%5D%2C%22package_license_set%22%3A%5B%5D%2C%22package_maintainers_set%22%3A%5B%5D%2C%22package_platforms%22%3A%5B%5D%7D&sort=relevance&type=packages&query=cudatoolkit number of different versions]. Please use the latest major version. You can see where they're defined in nixpkgs [https://github.com/NixOS/nixpkgs/blob/4d5b1d6b273fc4acd5dce966d2e9c0ca197b6df2/pkgs/development/compilers/cudatoolkit/default.nix here].
| |
| | |
| Several "CUDA-X" libraries are packages as well. In particular,
| |
| * cuDNN is packaged [https://github.com/NixOS/nixpkgs/blob/634141959076a8ab69ca2cca0f266852256d79ee/pkgs/development/libraries/science/math/cudnn/default.nix here].
| |
| * cuTENSOR is packaged [https://github.com/NixOS/nixpkgs/blob/634141959076a8ab69ca2cca0f266852256d79ee/pkgs/development/libraries/science/math/cutensor/default.nix here].
| |
| | |
| '''Note that these examples haven't been updated in a while (as of 2022-03-12). May not be the best solution. A better resource is likely the packaging CUDA sample code [https://github.com/NixOS/nixpkgs/blob/634141959076a8ab69ca2cca0f266852256d79ee/pkgs/test/cuda/cuda-library-samples/generic.nix here].'''
| |
| | |
| There some possible ways to setup a development environment using CUDA on NixOS. This can accomplished in the following ways:
| |
| | |
| * By making a FHS user env
| |
| | |
| {{file|cuda-fsh.nix|nix|<nowiki>
| |
| { pkgs ? import <nixpkgs> {} }:
| |
| | |
| let
| |
| fhs = pkgs.buildFHSUserEnv {
| |
| name = "cuda-env";
| |
| targetPkgs = pkgs: with pkgs; [
| |
| git
| |
| gitRepo
| |
| gnupg
| |
| autoconf
| |
| curl
| |
| procps
| |
| gnumake
| |
| utillinux
| |
| m4
| |
| gperf
| |
| unzip
| |
| cudatoolkit
| |
| linuxPackages.nvidia_x11
| |
| libGLU libGL
| |
| xorg.libXi xorg.libXmu freeglut
| |
| xorg.libXext xorg.libX11 xorg.libXv xorg.libXrandr zlib
| |
| ncurses5
| |
| stdenv.cc
| |
| binutils
| |
| ];
| |
| multiPkgs = pkgs: with pkgs; [ zlib ];
| |
| runScript = "bash";
| |
| profile = ''
| |
| export CUDA_PATH=${pkgs.cudatoolkit}
| |
| # export LD_LIBRARY_PATH=${pkgs.linuxPackages.nvidia_x11}/lib
| |
| export EXTRA_LDFLAGS="-L/lib -L${pkgs.linuxPackages.nvidia_x11}/lib"
| |
| export EXTRA_CCFLAGS="-I/usr/include"
| |
| '';
| |
| };
| |
| in pkgs.stdenv.mkDerivation {
| |
| name = "cuda-env-shell";
| |
| nativeBuildInputs = [ fhs ];
| |
| shellHook = "exec cuda-env";
| |
| }
| |
| </nowiki>}}
| |
| | |
| | |
| * By making a nix-shell
| |
| {{file|cuda-shell.nix|nix|<nowiki>
| |
| { pkgs ? import <nixpkgs> {} }:
| |
| | |
| pkgs.stdenv.mkDerivation {
| |
| name = "cuda-env-shell";
| |
| buildInputs = with pkgs; [
| |
| git gitRepo gnupg autoconf curl
| |
| procps gnumake utillinux m4 gperf unzip
| |
| cudatoolkit linuxPackages.nvidia_x11
| |
| libGLU libGL
| |
| xorg.libXi xorg.libXmu freeglut
| |
| xorg.libXext xorg.libX11 xorg.libXv xorg.libXrandr zlib
| |
| ncurses5 stdenv.cc binutils
| |
| ];
| |
| shellHook = ''
| |
| export CUDA_PATH=${pkgs.cudatoolkit}
| |
| # export LD_LIBRARY_PATH=${pkgs.linuxPackages.nvidia_x11}/lib:${pkgs.ncurses5}/lib
| |
| export EXTRA_LDFLAGS="-L/lib -L${pkgs.linuxPackages.nvidia_x11}/lib"
| |
| export EXTRA_CCFLAGS="-I/usr/include"
| |
| '';
| |
| }
| |
| </nowiki>}}
| |
| | |
| == See also ==
| |
| | |
| * [https://github.com/grahamc/nixos-cuda-example nixos-cuda-example]
| |
| * [https://github.com/grahamc/nixos-cuda-example/pull/2 nix-shell envs for Cuda]
| |
| * [https://discourse.nixos.org/t/cuda-setup-on-nixos/1118 CUDA setup on NixOS]
| |
| * [https://github.com/NixOS/nixpkgs/issues/131608 eGPU with nvidia-docker on intel-xserver]
| |
|
| |
|
| = Using your GPU for graphics = | | = Using your GPU for graphics = |