GPU 버전 TensorFlow CPU 실행 방법

309 단어 tensorflow
일부 원인으로 인해 GPU 버전의 TensorFlow가 실행되면 메모리가 넘치는 등 문제가 발생할 수 있습니다.이 때 우리는 CPU와 시스템 메모리로 프로그램을 실행할 수 있다.
코드는 다음과 같습니다.
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

상기 세 줄 코드를 프로그램의 첫 부분에 넣으면 된다.

좋은 웹페이지 즐겨찾기