Spark DataFrameに対してPandas UDFを使用してディープラーニング推論を行う際、GPUメモリ不足(OOM)エラーが発生した。チューニングすべきSparkの設定パラメータはどれか。

Pandas UDFに渡されるバッチサイズが大きすぎるとGPUメモリに収まらないため、`maxRecordsPerBatch`を小さく調整してバッチサイズを制御する。