Support for Ada Lovelace Architecture

NVIDIA / DeepLearningExamples

State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.

12.93k stars 3.12k forks source link

Support for Ada Lovelace Architecture #1371

Open ultrascale25 opened 5 months ago

ultrascale25 commented 5 months ago

I've just started exploring this deep learning repo but I noticed the repo says the supported GPUs are:

NVIDIA Volta
NVIDIA Turing
NVIDIA Ampere

What does this mean if I have an RTX 4090 running with the ada lovelace architecture, can I still train on my computer or with the docker container?