2 ms·
For anyone else using NixOS, https://github.com/nixos-rocm/nixos-rocm https://github.com/nixos-rocm/nixos-rocm by Anthony Cowley works very well. Not just NixO
by danieldk 7y ago
For anyone else using NixOS, https://github.com/nixos-rocm/nixos-rocm https://github.com/nixos-rocm/nixos-rocm by Anthony Cowley works very well.
Not just NixOS. Our work AMD GPU server uses Ubuntu 18.04, but I use Anthony's package set for easy use of PyTorch and Tensorflow in various environments (using nix-shell + direnv).