agentfuzzy999 t1_j6uzju1 wrote on February 2, 2023 at 1:51 AM

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.

AbCi16 OP t1_j6v0be2 wrote on February 2, 2023 at 1:57 AM

Command to check gpu availability?

agentfuzzy999 t1_j6v0d3e wrote on February 2, 2023 at 1:57 AM

Google it

AbCi16 OP t1_j6v0gql wrote on February 2, 2023 at 1:58 AM

👍

Some-Assistance-7812 t1_j6vwocu wrote on February 2, 2023 at 6:43 AM

import torch

torch.cuda.is_available()

cruddybanana1102 t1_j6wkzwb wrote on February 2, 2023 at 12:08 PM

If you have Nvidia with CUDA toolkit installed run nvidia-smi and you'll have your answer