caffe: use CUDA 10.1, cuDNN 7.6

The current version of caffe, v1.0, does not build with any later versions of cuDNN and CUDA 10.1 is the latest CUDA that is supported by cuDNN 7.6.
2022-04-03 03:07:59 +00:00
parent 2a0acdbe9f
commit 1ef351f4af
1 changed files with 3 additions and 0 deletions
@@ -33021,6 +33021,9 @@ with pkgs;
  ### SCIENCE / MATH

  caffe = callPackage ../applications/science/math/caffe ({
+    cudaSupport = config.cudaSupport or false;
+    cudatoolkit = cudatoolkit_10_1;
+    cudnn = cudnn_7_6_cudatoolkit_10_1;
    opencv3 = opencv3WithoutCuda; # Used only for image loading.
    blas = openblas;
    inherit (darwin.apple_sdk.frameworks) Accelerate CoreGraphics CoreVideo;