-
-
Notifications
You must be signed in to change notification settings - Fork 13.4k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
torch.compile
fails because it does not find ldconfig -p
#320854
Comments
Just in case this is the same error propagating from triton, could you please test #320888? I probably won't get around to do this myself for another week |
Hi @SomeoneSerge, I tested the PR you mentioned and it's surely a step in the right direction! I don't get the I get this now, which is technically a different problem but still related to
Do you know where this could stem from? For the curious, here's the flake I used to test the PR: {
description = "Check SomeoneSerge's ldconfig -p fix";
inputs.nixpkgs.url = "github:SomeoneSerge/nixpkgs/530d5769d57e88bde975c5fefb9ecfcddba89742";
outputs =
{ self, nixpkgs }:
let
pkgs = import nixpkgs {
system = "x86_64-linux";
config.allowUnfree = true;
};
in
{
devShell.x86_64-linux =
with pkgs;
(mkShell.override { stdenv = gcc11Stdenv; }) {
venvDir = "./.venv";
buildInputs = [
(pkgs.python3.withPackages (
ps: with ps; [
torch-bin
openai-triton-bin
]
))
pkgs.virtualenv
pkgs.python3Packages.venvShellHook
];
postVenvCreation = ''
unset SOURCE_DATE_EPOCH
'';
};
};
} |
Maybe our torch and triton had diverged? We may be relaxing version constraints sometimes and we may not have included integration tests for nixpkgs/pkgs/development/python-modules/torch/gpu-checks.nix Lines 16 to 26 in efcebae
nixpkgs/pkgs/development/python-modules/pynvml/default.nix Lines 55 to 60 in efcebae
|
Describe the bug
Trying to optimize a machine learning with
torch.compile
for a GPU creates an error becauseldconfig -p
is not found.Here's the precise error:
Steps To Reproduce
On a computer with
pytorch
and an enabled NVidia GPU, try to run the following python program:Expected behavior
The program should complete successfully.
Additional context
This reminds me of another issue I raised on not having
ldconfig -p
in apytorch
environment, #285307Notify maintainers
@NixOS/cuda-maintainers @SomeoneSerge @Madouura
Metadata
Please run
nix-shell -p nix-info --run "nix-info -m"
and paste the result.Add a 👍 reaction to issues you find important.
The text was updated successfully, but these errors were encountered: