(pytorch 번개 버그) ImportError: ‘pytorch_lightning.utilities.distributed’에서 이름 ‘rank_zero_only’를 가져올 수 없습니다(/opt/conda/lib/python3.8/site-packages/pytorch_lightning/utilities/distributed.py).


쉬운 목차

실수

ImportError: cannot import name 'rank_zero_only' from 'pytorch_lightning.utilities.distributed' (/opt/conda/lib/python3.8/site-packages/pytorch_lightning/utilities/distributed.py)

해결

rank_zero 함수의 위치가 변경되었기 때문입니다.

rank_zero를 가져온 위치를 변경하기만 하면 됩니다.

from pytorch_lightning.utilities.distributed import rank_zero_only
# 위 코드를 아래로 수정
from pytorch_lightning.utilities.rank_zero import rank_zero_only