Commit 7cd68c3d authored by Jonathan Ringer's avatar Jonathan Ringer Committed by Jonathan Ringer
Browse files

python3Packages.torch*: use pkgs.autoAddDriverRunpath

parent 65c4c21a

pkgs/development/python-modules/torch/bin.nix

+2 −1
Original line number Diff line number Diff line 
{ lib, stdenv

, buildPythonPackage

, autoAddDriverRunpath

, fetchurl

, python

, pythonAtLeast
@@ -40,7 +41,7 @@ in buildPythonPackage { 
  nativeBuildInputs = lib.optionals stdenv.isLinux [

    addOpenGLRunpath

    autoPatchelfHook

    cudaPackages.autoAddDriverRunpath

    autoAddDriverRunpath

  ];



  buildInputs = lib.optionals stdenv.isLinux (with cudaPackages; [

pkgs/development/python-modules/torch/default.nix

+1 −0
Original line number Diff line number Diff line 
{ stdenv, lib, fetchFromGitHub, fetchpatch, buildPythonPackage, python,

  config, cudaSupport ? config.cudaSupport, cudaPackages,

  autoAddDriverRunpath,

  effectiveMagma ?

  if cudaSupport then magma-cuda-static

  else if rocmSupport then magma-hip