Yes, you absolutely need to take Diamox on your Gokyo Lakes and Gokyo Ri trek. Diamox is taken as prevention for altitude sickness but not to cure its symptoms. After the symptoms begin to show, the only option you will have is to descend to a lower altitude.