python3Packages.cupy: cudatoolkit consistency

2 years ago · 2133acc806
parent 03dd3f66fb
commit 2133acc806
1 changed files with 4 additions and 0 deletions
--- a/pkgs/development/python-modules/cupy/default.nix
+++ b/pkgs/development/python-modules/cupy/default.nix
@ -5,6 +5,10 @@
 , addOpenGLRunpath
 }:

+assert cudnn.cudatoolkit == cudatoolkit;
+assert cutensor.cudatoolkit == cudatoolkit;
+assert nccl.cudatoolkit == cudatoolkit;
+
 buildPythonPackage rec {
  pname = "cupy";
  version = "10.2.0";