diff --git a/overrides/python/torch/default.nix b/overrides/python/torch/default.nix index 97676fd37..de37ad7b2 100644 --- a/overrides/python/torch/default.nix +++ b/overrides/python/torch/default.nix @@ -9,12 +9,14 @@ # use the autoAddOpenGLRunpathHook to add /run/opengl-driver/lib to the RPATH # of all ELF files deps = {nixpkgs, ...}: { - inherit (nixpkgs.cudaPackages) autoAddOpenGLRunpathHook; + inherit (nixpkgs.stdenv) isLinux; + autoAddOpenGLRunpathHook = lib.optionalAttribute nixpkgs.stdenv.isLinux nixpkgs.cudaPackages.autoAddOpenGLRunpathHook; }; - mkDerivation.nativeBuildInputs = [ + + mkDerivation.nativeBuildInputs = lib.mkIf config.deps.isLinux [ config.deps.autoAddOpenGLRunpathHook ]; # this file is patched manually, so ignore it in autoPatchelf - env.autoPatchelfIgnoreMissingDeps = ["libcuda.so.1"]; + env.autoPatchelfIgnoreMissingDeps = lib.mkIf config.deps.isLinux ["libcuda.so.1"]; }