본문 바로가기
파이썬

파이썬 TensorFlow가 GPU에 액세스하지 못하도록 차단 하시겠습니까?

by º기록 2020. 10. 17.
반응형

순수하게 CPU에서 TensorFlow를 실행하는 방법이 있습니까? 내 컴퓨터의 모든 메모리는 TensorFlow를 실행하는 별도의 프로세스에 의해 발생합니다. per_process_memory_fraction을 0으로 설정하려고 시도했지만 실패했습니다.

 

해결 방법

 


요약하면 다음 코드를 추가 할 수 있습니다.

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
import tensorflow as tf


export CUDA_VISIBLE_DEVICES = 를 원하거나 GPU가 아닌 TensorFlow 설치와 함께 virtualenv를 사용할 수도 있습니다.

 

참조 페이지 https://stackoverflow.com/questions/44552585

 

 

반응형

댓글