python super resolution InternalError 해결방법 좀 알려주세요

조회수 48회

안녕하세요. super resolution 학습시키려고 하는데 InternalError가 떠서 해결이 되지 않아 질문드립니다. 구글링 해보니까 이미지 크기가 너무 커서 그렇다고 하는데 이미지 크기를 줄일 수 없는 상황입니다. Input size는 2000*2000이고 데이터셋은 1000장 정도 됩니다. 사용하는 gpu는 NVIDIA RTX A6000 3개입니다.

Gpu 3개 다 돌렸을 때 학습이 될거라고 생각했는데 안돼서요 .. 혹시 해결 방법이 있을까요 ?

import os

os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"

os.environ["CUDA_VISIBLE_DEVICES"]= '0, 1, 2'

  • 원인이 이미지 크기가 맞아요? 이미지를 몇장 택해서 줄여서 수행해보고 확인하세요. 그냥 internall error 인데 글만 보곤 알수가 없습니다. 정영훈 2022.8.5 17:09

답변을 하려면 로그인이 필요합니다.

Hashcode는 개발자들을 위한 무료 QnA 사이트입니다. 계정을 생성하셔야만 답변을 작성하실 수 있습니다.

(ಠ_ಠ)
(ಠ‿ಠ)

ᕕ( ᐛ )ᕗ
로그인이 필요합니다

Hashcode는 개발자들을 위한 무료 QnA사이트 입니다. 계정을 생성하셔야만 글을 작성하실 수 있습니다.