By default, only one core is used for Yarn application master. To increase this, use the following command in the Spark shell:
val sc = new SparkContext(new SparkConf())
./bin/spark-submit <all your existing options> --spark.yarn.am.cores=<number of cores>