Hello,
The temperature of GPU is mainly related to GPU-Util. Can you check if both codes give you the same GPU-util and same memory usage on nvidia-smi ?
In case you see a difference in GPU-util, it maybe related to how you read and batch over the dataset, so you can first check the amount of time needed to read it batch in your tf code and torch code.