cafkafk/nixpkgs
cafkafk/nixpkgs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

python3Packages.torch*: use pkgs.autoAddDriverRunpath

Browse source
This commit is contained in:
Jonathan Ringer 2024-03-30 11:34:25 -07:00 committed by Jonathan Ringer
parent 65c4c21a2b
commit 7cd68c3dc8
2 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
pkgs/development/python-modules/torch/bin.nix
View file

@ -1,5 +1,6 @@
{ lib, stdenv { lib, stdenv
, buildPythonPackage , buildPythonPackage
, autoAddDriverRunpath
, fetchurl , fetchurl
, python , python
, pythonAtLeast , pythonAtLeast
@ -40,7 +41,7 @@ in buildPythonPackage {
nativeBuildInputs = lib.optionals stdenv.isLinux [ nativeBuildInputs = lib.optionals stdenv.isLinux [
addOpenGLRunpath addOpenGLRunpath
autoPatchelfHook autoPatchelfHook
cudaPackages.autoAddDriverRunpath autoAddDriverRunpath
]; ];
buildInputs = lib.optionals stdenv.isLinux (with cudaPackages; [ buildInputs = lib.optionals stdenv.isLinux (with cudaPackages; [

1
pkgs/development/python-modules/torch/default.nix
View file

@ -1,5 +1,6 @@
{ stdenv, lib, fetchFromGitHub, fetchpatch, buildPythonPackage, python, { stdenv, lib, fetchFromGitHub, fetchpatch, buildPythonPackage, python,
config, cudaSupport ? config.cudaSupport, cudaPackages, config, cudaSupport ? config.cudaSupport, cudaPackages,
autoAddDriverRunpath,
effectiveMagma ? effectiveMagma ?
if cudaSupport then magma-cuda-static if cudaSupport then magma-cuda-static
else if rocmSupport then magma-hip else if rocmSupport then magma-hip