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