Open aryanjassal opened 2 months ago
Can you see if:
pkgs.python3Packages.tensorflow-bin
workspkgs.python3Packages.tensorflow-bin_2
works (actually this is TF2 right?)You need to activate the cudaSupport
to be true to get the actual GPU activation.
I suspect tensorflowWithCuda
is just broken. A few years back there was no need to provide all those environment variables, the package should just work, if it doesn't I don't think the package maintainers have done it correctly. I reckon there should be some sort of shell hook that automatically sets up everything that is required so one can start hacking in a Python 3 shell immediately without messing with random env variables.
I have tried using both tensorflowWithCuda
and tensorflow-bin
and nothing worked. tensorflowWithCuda
failed to compile altogether, throwing a hash mismatch error when building through bazel, and tensorflow-bin
just couldn't find or load CUDA drivers.
I have ensured that my system is set up correctly, as running pytorchWithCuda
works perfectly and can also detect my GPU without issues. I have also tried running a demo script and used nvidia-smi
to confirm that my GPU is, in fact, being used.
Steps To Reproduce
Steps to reproduce the behavior:
python311Packages.tensorflowWithCuda
Build log
https://gist.github.com/aryanjassal/cb19d4d335743504dd35404d153ed663
Additional context
My
flake.nix
file which compiles this as a dependency:Previously, I was defining tensorflow as follows:
But on some revisions it would error out with the following log:
So, I presumed that it could not find some CUDA libraries, so I provided the
cudatoolkit
path onCUDA_PATH
,EXTRA_CCFLAGS
, andEXTRA_LDFLAGS
, which let me avoid that error.The output of
nix flake metadata
, which shows thenixpkgs
andflake-utils
revisions.Notify maintainers
@abbradar
Metadata
Please run
nix-shell -p nix-info --run "nix-info -m"
and paste the result.Add a :+1: reaction to issues you find important.