gador/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #329511 from trofi/python3Packages.pytorch-bin.tests-fix-eval

Browse Source 
python3Packages.pytorch-bin.tests: fix the eval
This commit is contained in:
Someone
2024-07-24 00:41:14 +03:00
committed by GitHub
parent af6746331d f84f1c2744
commit 8cdf7e751d
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/python-modules/torch/bin.nix
+1 -4
View File
@@ -121,10 +121,7 @@ buildPythonPackage {
pythonImportsCheck = [ "torch" ];
passthru.tests = callPackage ./tests.nix {
torchWithCuda = torch-bin;
torchWithRocm = torch-bin;
};
passthru.tests = callPackage ./tests.nix {};
meta = {
description = "PyTorch: Tensors and Dynamic neural networks in Python with strong GPU acceleration";