cudatoolkit: install missing libraries for nvprof (10.1 and 10.2)

4 years ago · d2a6b1d55a
parent 58b66c0dcb
commit d2a6b1d55a
1 changed files with 2 additions and 1 deletions
--- a/pkgs/development/compilers/cudatoolkit/common.nix
+++ b/pkgs/development/compilers/cudatoolkit/common.nix
@ -154,8 +154,9 @@ stdenv.mkDerivation rec {

    # Remove OpenCL libraries as they are provided by ocl-icd and driver.
    rm -f $out/lib64/libOpenCL*
-    ${lib.optionalString (lib.versionAtLeast version "10.1") ''
+    ${lib.optionalString (lib.versionAtLeast version "10.1" && (lib.versionOlder version "11")) ''
      mv $out/lib64 $out/lib
+      mv $out/extras/CUPTI/lib64/libcupti* $out/lib
    ''}

    # Set compiler for NVCC.