Hello
I've got a new laptop recently and it has NVIDIA GeoForce RTX 3070 GPU. I've installed CUDA as well and deep learning packages. Everything is fine when I run my code on CPU but when I switch to GPU, it just provides faulty results. It happens in prediction but I realised even for training, it doesn't actually run anything on GPU.
import torch
torch.cuda.is_available()
When I run this, it says true. How can I solve this? I'm using ArcGIS Pro 2.8.2 and had the same experience using 2.7 on this laptop.
