cafkafk/nixpkgs
cafkafk/nixpkgs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

Merge pull request #170786 from SomeoneSerge/chainer

Browse source 
chainer: fix cuda
This commit is contained in:
Sandro 2022-05-09 13:30:41 +02:00 committed by GitHub
commit 1f08cf0879
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
pkgs/development/python-modules/chainer/default.nix
View file

@ -30,6 +30,11 @@ buildPythonPackage rec {
pytestFlagsArray = [ "tests/chainer_tests/utils_tests" ];
# cf. https://github.com/chainer/chainer/issues/8621
preCheck = ''
export CHAINER_WARN_VERSION_MISMATCH=0
'';
disabledTests = [
"gpu"
"cupy"
@ -39,8 +44,6 @@ buildPythonPackage rec {
meta = with lib; {
description = "A flexible framework of neural networks for deep learning";
homepage = "https://chainer.org/";
# Un-break me when updating chainer next time!
broken = cudaSupport && (lib.versionAtLeast cupy.version "8.0.0");
license = licenses.mit;
maintainers = with maintainers; [ hyphon81 ];
};