update
This commit is contained in:
parent
4bab842687
commit
347d5c6554
1 changed files with 5 additions and 3 deletions
|
|
@ -29,15 +29,17 @@
|
|||
packages."${system}" = {
|
||||
# just one main.cu file for now
|
||||
# compile it and place it to $out/bin/cuda-kernel
|
||||
cuda-kernel = pkgs.stdenv.mkDerivation {
|
||||
cuda-kernel = pkgs.cudaPackages.backendStdenv.mkDerivation {
|
||||
name = "cuda-kernel";
|
||||
src = ./.;
|
||||
buildInputs = with pkgs; [
|
||||
cudatoolkit
|
||||
cudaPackages.cuda_cudart
|
||||
linuxPackages.nvidia_x11
|
||||
linuxPackages.nvidia_x11_beta
|
||||
];
|
||||
LD_LIBRARY_PATH = "${pkgs.linuxPackages.nvidia_x11}/lib";
|
||||
cudaSupport = true;
|
||||
requiredSystemFeatures = [ "cuda" ];
|
||||
LD_LIBRARY_PATH = "${pkgs.linuxPackages.nvidia_x11_beta}/lib";
|
||||
buildPhase = ''
|
||||
nvcc -o main main.cu
|
||||
mkdir -p $out/bin
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue