안녕하세요. 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'