python super resolution InternalError 해결방법 좀 알려주세요
조회수 335회
안녕하세요. 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'
댓글 입력