AttributeError: module 'torch.distributed' has no attribute '_reduce_scatter_base'

boingboing·2024년 3월 12일
  • apex 패키지 사용할 때 발생한 에러.

  • 구글링 결과, torch >= 1.10 이상이어야 해당 에러가 발생 안 함.

  • torch GPU는 torch_cuda만 쓰는거 아닌가..? 라고 생각했는데 그게 아님.

참고)
https://velog.io/@sandartchip/torch%EC%99%80-torchcuda

0개의 댓글