I deployed my trained custom model to Vertex AI and used the GEE API to call the model for predicting images on Earth Engine. However, I encountered the following error message:
"EEException: Vertex AI prediction error: 2 root error(s) found. (0) UNIMPLEMENTED: Could not find compiler for platform CUDA: NOT_FOUND: could not find registered compiler for platform CUDA -- was support for that platform linked in? [[{{function_node __inference__wrapped_model_13951}}{{node model_1/UModel_with_reshaping/UModel/down_blocEEException: Vertex AI prediction error: XLA compilation disabled [[{{function_node __inference__wrapped_model_16067}}{{node model/my_model/UModel/down_blocks0_nafBlock/sequential_2/down_blocks0_nafBlock_conv_2/PartitionedCall}}]]ks0_nafBlock/sequential_2/down_blocks0_nafBlock_conv_2/PartitionedCall}}]] [[StatefulPartitionedCall/StatefulPartitionedCall/model_1/re_serialize_output_2/map/while/body/_360/model_1/re_serialize_output_2/map/while/TensorArrayV2Read/TensorListGetItem/_356]] (1) UNIMPLEMENTED: Could not find compiler for platform CUDA: NOT_FOUND: could not find registered compiler for platform CUDA -- was support for that platform linked in? [[{{function_node __inference__wrapped_model_13951}}{{node model_1/UModel_with_reshaping/UModel/down_blocks0_nafBlock/sequential_2/down_blocks0_nafBlock_conv_2/PartitionedCall}}]] 0 successful operations. 0 derived errors ignored."
I change CONTAINER_IMAGE from “us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-11:latest” to “us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-13:latest”，than I got an new error :
EEException: Vertex AI prediction error: XLA compilation disabled [[{{function_node __inference__wrapped_model_16067}}{{node model/my_model/UModel/down_blocks0_nafBlock/sequential_2/down_blocks0_nafBlock_conv_2/PartitionedCall}}]]
Can you please advise on how to resolve this issue?
LinkedIn
Twitter
