[tensorflow] tensorflow를 multiprocessing으로 실행시킬 경우

최승언·2022년 11월 11일
0

tensorflow

목록 보기
1/1

에러코드
[error] Attempting to perform BLAS operation using StreamExecutor without BLAS support

tensorflow-gpu를 설치하고 predict함수를 multiprocessing으로 돌릴경우 발생하는 것으로 보인다. 간단하게 말해 tensorflow를 2개이상 동시에 실행할 경우 발생하는 에러이다.
간단하게 코드에 아래 내용을 추가하자.

import tensorflow as tf
physical_devices = tf.config.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(physical_devices[0], True)

단 여기서 set_memory_growth 함수는 메모리 증가를 활성화하는 함수로 gpu 메모리 크기에 따라 실행에 문제가 생길 수 있다는 점을 주의해야 한다.

profile
작업하다가 막힌부분을 기록하는 곳.

0개의 댓글