pyrox.dev / nixpkgs
lol
fork atom

python3Packages.torch*: use pkgs.autoAddDriverRunpath

authored by Jonathan Ringer and committed by Jonathan Ringer 7cd68c3d 65c4c21a

+3 -1
unified split
+2 -1
pkgs/development/python-modules/torch/bin.nix
··· 1 { lib, stdenv 2 , buildPythonPackage 3 , fetchurl 4 , python 5 , pythonAtLeast ··· 40 nativeBuildInputs = lib.optionals stdenv.isLinux [ 41 addOpenGLRunpath 42 autoPatchelfHook 43 - cudaPackages.autoAddDriverRunpath 44 ]; 45 46 buildInputs = lib.optionals stdenv.isLinux (with cudaPackages; [
··· 1 { lib, stdenv 2 , buildPythonPackage 3 + , autoAddDriverRunpath 4 , fetchurl 5 , python 6 , pythonAtLeast ··· 41 nativeBuildInputs = lib.optionals stdenv.isLinux [ 42 addOpenGLRunpath 43 autoPatchelfHook 44 + autoAddDriverRunpath 45 ]; 46 47 buildInputs = lib.optionals stdenv.isLinux (with cudaPackages; [
+1
pkgs/development/python-modules/torch/default.nix
··· 1 { stdenv, lib, fetchFromGitHub, fetchpatch, buildPythonPackage, python, 2 config, cudaSupport ? config.cudaSupport, cudaPackages, 3 effectiveMagma ? 4 if cudaSupport then magma-cuda-static 5 else if rocmSupport then magma-hip
··· 1 { stdenv, lib, fetchFromGitHub, fetchpatch, buildPythonPackage, python, 2 config, cudaSupport ? config.cudaSupport, cudaPackages, 3 + autoAddDriverRunpath, 4 effectiveMagma ? 5 if cudaSupport then magma-cuda-static 6 else if rocmSupport then magma-hip