컴파일러는 컴파일하는 동안 자동으로 GPU를 사용합니까, 아니면 수동으로 사용해야합니까?

Question

나는 8GB RAM, 인텔 중핵 i7-4th gen를 가진 나의 PC에 bird image preprocessing를위한이 프로그램을 집계하고있다. 프로세서하지만 매번 IS이 오류를 표시하는 동안 실행 한 후 내동댕이 :

File "bird.py", line 13, in <module> 
    X,Y,X_test,Y_test = pickle.load(open("full_dataset.pkl", "rb")) 
MemoryError 

이 tensorflow하는 초보자 인을, 나는 CUDA는 (I 성공적으로 설치 한 함) 사용 여부를 확실하지 않다거나 단지입니다 CPU는 여전히 계산을 수행합니다.

Answer 1

보고있는 오류는 GPU 계산 (또는 실제로 TensorFlow 계산)이 실행되기 전에 발생합니다. MemoryError은 데이터 집합 unpickle이 메모리 부족으로 이어 지거나 데이터에 손상이 있음을 의미합니다. 작은 데이터 세트를로드하여 문제가 지속되는지 확인하십시오.

원래 질문에 대해서는 GPU 버전의 TensorFlow와 호환되는 CUDA 버전 (예 : CUDA 8.0) 및 cuDNN을 설치 한 경우 TensorFlow가 행렬 곱셈 및 훈련 중에 소비 된 대부분의 시간을 구성하는 회선.