인프런 커뮤니티 질문&답변
작성한 질문수
답변 1
1
AISchool
지식공유자
안녕하세요~. 반갑습니다.
해당코드는 GPU 관련 문제인데요.
여러 원인이 있을수 있어서 우선 코드 상단에 아래 코드를 추가해서 tensorflow를 CPU 모드로 실행하시는것을 추천드립니다. (GPU 사용을 중지하는 코드입니다.)
import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="-1"
감사합니다.