File tree Expand file tree Collapse file tree 1 file changed +18
-11
lines changed Expand file tree Collapse file tree 1 file changed +18
-11
lines changed Original file line number Diff line number Diff line change @@ -67,16 +67,23 @@ stdenv'.mkDerivation rec {
6767 boost
6868 ]
6969 ++ lib . optionals ( backend == "eigen" ) [ eigen ]
70- ++ lib . optionals ( backend == "cuda" ) [
71- cudaPackages . cuda_cudart
72- cudaPackages . cudnn
73- cudaPackages . cudatoolkit
74- ]
75- ++ lib . optionals ( backend == "tensorrt" ) [
76- cudaPackages . cuda_cudart
77- cudaPackages . cudatoolkit
78- cudaPackages . tensorrt
79- ]
70+ ++ lib . optionals ( backend == "cuda" ) (
71+ with cudaPackages ;
72+ [
73+ cuda_cccl
74+ cuda_cudart
75+ cuda_nvcc
76+ cudnn
77+ libcublas
78+ ]
79+ )
80+ ++ lib . optionals ( backend == "tensorrt" ) (
81+ with cudaPackages ;
82+ [
83+ cuda_cudart
84+ tensorrt
85+ ]
86+ )
8087 ++ lib . optionals ( backend == "opencl" ) [
8188 opencl-headers
8289 ocl-icd
@@ -103,7 +110,7 @@ stdenv'.mkDerivation rec {
103110 cd cpp/
104111 ''
105112 + lib . optionalString ( backend == "cuda" || backend == "tensorrt" ) ''
106- export CUDA_PATH="${ cudaPackages . cudatoolkit } "
113+ export CUDA_PATH="${ cudaPackages . cuda_nvcc } "
107114 export EXTRA_LDFLAGS="-L/run/opengl-driver/lib"
108115 '' ;
109116
You can’t perform that action at this time.
0 commit comments