Он входил в состав Мали U-20 команды, занявшие третье место в групповом этапе молодёжного чемпионата мира 2003 года. Диаките входил в состав футбольной олимпийской сборной Мали 2004, которая вышла в четвертьфинал, заняв 1-ое место в группы А, но проиграли Италии в следующем раунде. Во взрослой сборной Диаките дебютировал в официальных матчах в 2006 году. Со основной сборной Мали он принимал участие в шести Кубках африканских наций (2008—2017), где дважды сумел завоевать бронзовые медали турнира в 2012 и 2013 годах. За сборную Мали он принял участие в 51 матче.