When I click on the "Convert Unet to ONNX" button in the "Convert to ONNX" tab, I am prompted with the following error
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 mat1 in method wrapper_CUDA_addmm
How can I solve this problem?
When I click on the "Convert Unet to ONNX" button in the "Convert to ONNX" tab, I am prompted with the following error
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 mat1 in method wrapper_CUDA_addmm
How can I solve this problem?