딥러닝으로 이미지 분류해보고 있는데,,,


이미지들 데이터셋에 로드하고 tf.data.Dataset.from_tensor_slices() 메소드로 tf형식으로 변환시키면 


InternalError: Failed copying input tensor from /job:localhost/replica:0/task:0/device:CPU:0 to /job:localhost/replica:0/task:0/device:GPU:0 in order to run _EagerConst: Dst tensor is not initialized.

위처럼 에러 뜨면서 안되는데,,,gpu 메모리가 부족해서 인거 같아서 오류 해결하는 방법 계속 찾아봤는데도 해결방법이 도무지 안나오네요ㅠㅠㅠ


혹시 어떻게 하면 해결할 수 있을지,,,,,,,