If that's needed idk but installed as pointed out here https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=WSL-Ubuntu&target_version=2.0&target_type=deb_local
Also I have this config running: https://github.com/573/nix-config-1/tree/43f2a1453e7811489a90d95fedd1ced786768067/hosts/DANIELKNB1 and https://github.com/573/nix-config-1/blob/43f2a1453e7811489a90d95fedd1ced786768067/nixos/base/general.nix#L127 and https://github.com/573/nix-config-1/blob/43f2a1453e7811489a90d95fedd1ced786768067/home/base/general.nix#L116
Test with https://github.com/573/nix-config-1/blob/43f2a1453e7811489a90d95fedd1ced786768067/wsl-nixos-cuda-test/shell.nix (via https://discourse.nixos.org/t/pytorch-cuda-on-wsl/18267/2)
Finally did this https://stackoverflow.com/questions/48152674/how-do-i-check-if-pytorch-is-using-the-gpu
Last active
March 3, 2025 01:22
-
-
Save 573/8533a3a7a4b388f582b1cfed89f3e981 to your computer and use it in GitHub Desktop.
nix tip: use gpu nvidia on wsl2 nixos
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment