pytorch

1、pytorch

• conda install conda-forge::pytorch-gpu
  
• torch.utils.data.DataLoader
  
• torchvision.transforms.v2
  

2、pytorch_lightning

3、log

You are using a CUDA device ('NVIDIA GeForce RTX 3060 Laptop GPU') that has Tensor Cores. 

To properly utilize them, you should set `torch.set_float32_matmul_precision('medium' | 'high')` which will trade-off precision for performance. 

For more details, read https://pytorch.org/docs/stable/generated/torch.set_float32_matmul_precision.html#torch.set_float32_matmul_precision