Deep learning frameworks are not running on GPU

174
2
11-18-2021 05:48 PM
MaryamBarzegar
New Contributor II

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.

0 Kudos
2 Replies
DanPatterson
MVP Esteemed Contributor

ArcGIS Pro 2.9 system requirements—ArcGIS Pro | Documentation

from there, a suggestion is to ensure that you have the latest drivers for your setup.  Can you confirm that is the case?


... sort of retired...
0 Kudos
MaryamBarzegar
New Contributor II

@DanPatterson  I have the latest drivers too

0 Kudos