nixpkgs/overlays/torch.nix

   1 finalPkgs: prevPkgs: {
   2   python3Packages =
   3     (prevPkgs.python3.override {
   4       #self = finalPkgs.python3Packages;
   5       packageOverrides = finalPython: prevPython: {
   6         torch-bin = finalPython.torch;
   7         torchvision-bin = prevPython.torchvision;
   8         #tensorflow-bin = prevPython.tensorflow;
   9         torch = prevPython.torch.override {
  10           triton = prevPython.triton-no-cuda;
  11           rocmSupport = false;
  12           cudaSupport = false;
  13         };
  14       };
  15     }).pkgs;
  16 }