Submitted by AbCi16 t3_10rcty9 in deeplearning

I am new to deep learning. Till now I was mostly doing stuff on Kaggle. But now I am planning to to do stuff on Jupyter via GPU. But I have no idea how to do it. I read somewhere that I need Docker to do it. But I have never used Docker before. Should I install Docker Desktop or is their any other way to set it up?

1

Comments

You must log in or register to comment.

agentfuzzy999 t1_j6uzju1 wrote

You do not need docker. Open a notebook, import torch or tensorflow, check if GPU is available. If true, profit. If false, you have python/framework/CUDA problems.

5

FastestLearner t1_j6v8nsz wrote

Being able to use the GPU doesn’t have anything to do with Jupyter. It’s the packages (TensorFlow, PyTorch, etc.) that must be installed with CUDA support and also you must have the correct drivers installed. My recommendation would be simply use a conda environment, which automatically installs the correct CUDA packages during a PyTorch install or a Tensorflow install.

8

Appropriate_Ant_4629 t1_j6vmpm8 wrote

> Being able to use the GPU doesn’t have anything to do with Jupyter.

It's certainly not required....

.. but Nvidia makes it extremely convenient through the notebooks they provide:

https://catalog.ngc.nvidia.com/resources

>> The NGC catalog offers step-by-step instructions and scripts through Jupyter Notebooks for various use cases, including machine learning, computer vision, and conversational AI. These resources help you examine, understand, customize, test, and build AI faster, while taking advantage of best practices.

4

Some-Assistance-7812 t1_j6w1l5z wrote

Yeah it’s CUDA capable! RTX is much better than GTX for deep learning, and 2080 super is powerful! I did plenty of deep learning on my laptop using 1050ti using PyTorch. If you need any help, we can connect and help you out on Zoom call. It won’t take more than 20-30 mins.

2

AwkwardlyPure t1_j6w7x6m wrote

What other Python packages do I need to install ? Sometimes I get a warning about not having Tensorrt, then when I install it shows version 0.0.1 but apparently it's at version 8 ? There is a guide on nvidias website with some snippets of code but I don't fully understand.

I have already install TensorFlow and Keras.

1