[PyTorch] ValueError: prefetch_factor option could only be specified in multiprocessing.let num_workers > 0 to enable multiprocessing. 해결법

환길·2024년 4월 15일

Error

목록 보기
6/13

오류


ValueError: prefetch_factor option could only be specified in multiprocessing.let num_workers > 0 to enable multiprocessing.

다음과 같은 오류는 pytorch=1.x.x 버전에서 data를 Trainer에 불러오면서 DataLoader와 Dataset의 버전 문제로 인해 발생합니다.

이 때는 pytorch를 2.x.x 버전을 설치하면 해결됩니다.

해결법


pip install torch==2.0.0

다음과 같이 커맨드창에 입력하면 해결됩니다.

0개의 댓글