Commit 1c92ae96 authored by Martin Schwaighofer's avatar Martin Schwaighofer
Browse files

python3Packages.torchWithRocm: switch back to ROCm 5.7

parent edeff1bd

pkgs/development/python-modules/torch/default.nix

+3 −1
Original line number Diff line number Diff line
@@ -52,7 +52,7 @@ 


  # ROCm dependencies

  rocmSupport ? config.rocmSupport,

  rocmPackages,

  rocmPackages_5,

  gpuTargets ? [ ]

}:
@@ -60,6 +60,8 @@ let 
  inherit (lib) attrsets lists strings trivial;

  inherit (cudaPackages) cudaFlags cudnn nccl;



  rocmPackages = rocmPackages_5;



  setBool = v: if v then "1" else "0";



  # https://github.com/pytorch/pytorch/blob/v2.0.1/torch/utils/cpp_extension.py#L1744