Den 21. september 2017 ble Chelsea og Atlético Madrid enige om en overgangssum for Diego Costa. 31. desember 2017 skrev han under for Atlético Madrid. Han får draktnummer 18.