b12f/nixpkgs
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Merge pull request #217330 from ConnorBaker/fix/caffe-use-pre-cudnn8

Browse source 
caffe: use pre-CUDNN 8.x
This commit is contained in:
Samuel Ainsworth 2023-02-22 12:53:26 -05:00 committed by GitHub
parent fe1162da90 291db05669
commit 2308c637ef
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
pkgs/applications/science/math/caffe/default.nix
View file

@ -21,7 +21,13 @@
}:
let
inherit (cudaPackages) cudatoolkit cudnn nccl;
inherit (cudaPackages) cudatoolkit nccl;
# The default for cudatoolkit 10.1 is CUDNN 8.0.5, the last version to support CUDA 10.1.
# However, this caffe does not build with CUDNN 8.x, so we use CUDNN 7.6.5 instead.
# Earlier versions of cudatoolkit use pre-8.x CUDNN, so we use the default.
cudnn = if lib.versionOlder cudatoolkit.version "10.1"
then cudaPackages.cudnn
else cudaPackages.cudnn_7_6_5;
in
assert leveldbSupport -> (leveldb != null && snappy != null);