Hello,
I have Pytorch code written for semantic segmentation, technically code is working fine without any errors. But I am facing High GPU temperature issue. Same code is written in Tensorflow, My GPU only reach 65 °C temperature, But in Pytorch case my GPU reach 85 °C temperature. Here is all details of my GPU and training configurations:
Hardware:
GPU: NVIDIA GeForce RTX 2060 Super - 8GB
RAM: 16 GB
Software:
Pytorch v1.6
CUDA 10.2
Training Setting:
Img_size = (256, 256)
batch_size = 4
num_epochs = 100
lr = 1e-4
num_workers=2