3
0
Fork 0
forked from mirrors/nixpkgs

python310Packages.mxnet: mark cudaSupport broken

This commit is contained in:
Someone Serge 2022-04-05 19:02:03 +03:00
parent b1674bf9a5
commit 565201d769
No known key found for this signature in database
GPG key ID: A39079F4AE78D20D
2 changed files with 11 additions and 1 deletions

View file

@ -62,6 +62,12 @@ stdenv.mkDerivation rec {
rm "$out"/lib/*.a
'';
# used to mark cudaSupport in python310Packages.mxnet as broken;
# other attributes exposed for consistency
passthru = {
inherit cudaSupport cudnnSupport cudatoolkit cudnn;
};
meta = with lib; {
description = "Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler";
homepage = "https://mxnet.incubator.apache.org/";

View file

@ -6,10 +6,11 @@
, graphviz
, python
, isPy3k
, isPy310
}:
buildPythonPackage {
inherit (pkgs.mxnet) pname version src meta;
inherit (pkgs.mxnet) pname version src;
buildInputs = [ pkgs.mxnet ];
propagatedBuildInputs = [ requests numpy graphviz ];
@ -32,4 +33,7 @@ buildPythonPackage {
ln -s ${pkgs.mxnet}/lib/libmxnet.so $out/${python.sitePackages}/mxnet
'';
meta = pkgs.mxnet.meta // {
broken = (pkgs.mxnet.broken or false) || (isPy310 && pkgs.mxnet.cudaSupport);
};
}