let
  pkgs = import <nixpkgs> {};
  gcc7Stdenv = pkgs.gcc7Stdenv;

in gcc7Stdenv.mkDerivation (with pkgs; {
  name = "env";
  nativeBuildInputs = [
    cmake
    linuxPackages.nvidia_x11
    makeWrapper
    cudatoolkit
    cudnn
  ];

  buildInputs = [
    libzip
    boost
  ];

  shellHook = ''
    export CUDA_PATH=${pkgs.cudatoolkit}
    export EXTRA_LDFLAGS="-L/lib -L${pkgs.linuxPackages.nvidia_x11}/lib"
    export EXTRA_CCFLAGS="-I/usr/include"
  '';
})