search

NVIDIA-AI-IOT / jetbot

An educational AI robot based on NVIDIA Jetson Nano.
MIT License
3k stars 1.03k forks source link

Where can I find how to setup the GPU enabled training PC? #326

Closed HyperStudio-2020 closed 2 years ago

tokk-nv commented 3 years ago

Hello HyperStudeio-2020,

The easiest way to set up a Linux training machine is probably to go with Docker on your Linux PC. This document walks you through the general setup of Docker on your GPU enabled PC. https://docs.nvidia.com/deeplearning/frameworks/user-guide/index.html

There is a PyTorch docker container available on NGC, but you may need to install Jupyter environment by yourself. https://ngc.nvidia.com/catalog/containers/nvidia:pytorch

HyperStudio-2020 commented 3 years ago

Hey!

Thanks for the advice, is there any way I can set this up and train on my windows PC, I have an RTX 2060, so that's fine for training. But I really dont know how to train a model, just as the COCO model for example. How can I get my custom .engine file?

Thanks for reading! Hope to hear from you soon!

With best regards,

Thom

http://www.avg.com/email-signature?utm_medium=email&utm_source=link&utm_campaign=sig-email&utm_content=webmail Virus-free. www.avg.com http://www.avg.com/email-signature?utm_medium=email&utm_source=link&utm_campaign=sig-email&utm_content=webmail <#DAB4FAD8-2DD7-40BB-A1B8-4E2AA1F9FDF2>

Op ma 30 nov. 2020 om 19:12 schreef Chitoku YATO notifications@github.com:

Hello HyperStudeio-2020,

The easiest way to set up a Linux training machine is probably to go with Docker on your Linux PC. This document walks you through the general setup of Docker on your GPU enabled PC. https://docs.nvidia.com/deeplearning/frameworks/user-guide/index.html

There is a PyTorch docker container available on NGC, but you may need to install Jupyter environment by yourself. https://ngc.nvidia.com/catalog/containers/nvidia:pytorch

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub https://github.com/NVIDIA-AI-IOT/jetbot/issues/326#issuecomment-735952604, or unsubscribe https://github.com/notifications/unsubscribe-auth/AKIRO2TE3LMHUUJ4COCETB3SSPOAHANCNFSM4T7NXYFA .