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

Merge pull request #113223 from danieldk/pytorch-fixes

Browse Source 
python3Packages.pytorch-bin: do not depend on nvidia_x11
This commit is contained in:
Daniël de Kok
2021-02-17 09:05:21 +01:00
committed by GitHub
parent 58674cfe7e ebf348fc9a
commit c58a900f61
2 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/python-modules/pytorch/bin.nix
+1 -2
View File
@@ -5,7 +5,6 @@
, isPy38 , isPy38
, isPy39 , isPy39
, python , python
, nvidia_x11
, addOpenGLRunpath , addOpenGLRunpath
, future , future
, numpy , numpy
@@ -52,7 +51,7 @@ in buildPythonPackage {
''; '';
postFixup = let postFixup = let
rpath = lib.makeLibraryPath [ stdenv.cc.cc.lib nvidia_x11 ]; rpath = lib.makeLibraryPath [ stdenv.cc.cc.lib ];
in '' in ''
find $out/${python.sitePackages}/torch/lib -type f \( -name '*.so' -or -name '*.so.*' \) | while read lib; do find $out/${python.sitePackages}/torch/lib -type f \( -name '*.so' -or -name '*.so.*' \) | while read lib; do
echo "setting rpath for $lib..." echo "setting rpath for $lib..."
pkgs/top-level/python-packages.nix
+1 -3
View File
@@ -6505,9 +6505,7 @@ in {
pytorch = callPackage ../development/python-modules/pytorch { cudaSupport = pkgs.config.cudaSupport or false; }; pytorch = callPackage ../development/python-modules/pytorch { cudaSupport = pkgs.config.cudaSupport or false; };
pytorch-bin = callPackage ../development/python-modules/pytorch/bin.nix { pytorch-bin = callPackage ../development/python-modules/pytorch/bin.nix { };
inherit (pkgs.linuxPackages) nvidia_x11;
};
pytorch-lightning = callPackage ../development/python-modules/pytorch-lightning { }; pytorch-lightning = callPackage ../development/python-modules/pytorch-lightning { };