JAX: Difference between revisions
imported>Samuela No edit summary |
imported>Samuela No edit summary |
||
| Line 24: | Line 24: | ||
hardware.opengl.enable = true; | hardware.opengl.enable = true; | ||
</syntaxHighlight> | </syntaxHighlight> | ||
Then you can use the | Then you can use the <code>jaxlibWithCuda</code> package (equivalent to setting the <code>cudaSupport</code> parameter): | ||
<syntaxHighlight lang=nix> | <syntaxHighlight lang=nix> | ||
let | let | ||
pkgs = import (fetchTarball("https://github.com/NixOS/nixpkgs/archive/eac6215607e4ccceb9631b01ee8f8903a6e82e02.tar.gz")) {}; | pkgs = import (fetchTarball("https://github.com/NixOS/nixpkgs/archive/eac6215607e4ccceb9631b01ee8f8903a6e82e02.tar.gz")) {}; | ||
in pkgs.mkShell { | in pkgs.mkShell { | ||