webui로 사용하고 있구요
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument index in method wrapper_CUDA__index_select)
이렇게 나옵니다.
고수님들 알려주시면 감사하겠습니다. ㅠㅠ
연산이 cpu였다가 gpu였다가 왔다리 갔다리 한다는 뜻같은데 cpu에서만 동작되게 하든지 gpu에서만 동작되게 하던지 해야될듯?