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

This commit is contained in:
Thomas Bereknyei 2020-03-24 10:13:04 -04:00 committed by Daniël de Kok
parent 58b66c0dcb
commit d2a6b1d55a

View File

@ -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.