blob: c205d68868e2400d22e94f269d6b149684d3a5b8 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
{ pkgs ? import <nixpkgs> { }, ... }:
pkgs.stdenvNoCC.mkDerivation rec {
name = "boltzgen-env";
env = pkgs.buildEnv { name = name; paths = buildInputs; };
buildInputs = let
boltzgen = (import ../../boltzgen.nix { }).boltzgen pkgs.python3;
local-python = pkgs.python3.withPackages (python-packages: with python-packages; [
boltzgen
]);
in with pkgs; [
local-python
cmake
gcc
cudatoolkit
linuxPackages.nvidia_x11
];
shellHook = ''
export NIX_SHELL_NAME="${name}"
export CUDA_PATH="${pkgs.cudatoolkit}"
export PYTHONPATH="$PWD:$PYTHONPATH"
'';
}
|