This is specifically called out as a reason in the Hiroshima Peace Memorial Museum. I was there just a few weeks ago. It's also available here:
https://hpmmuseum.jp/modules/exhibition/index.php?action=Cor...
"The United States believed that ending the war with an atomic bombing would help prevent the Soviet Union from extending its sphere of influence."