You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
if the framework is pytorch, it will from donkeycar.parts.pytorch.torch_train import train and use it to train.
in this train function, it has following codes
iftorch.cuda.is_available():
print('Using CUDA')
gpus=-1else:
print('Not using CUDA')
gpus=0
however, when creating the trainer, it uses cpu flag and never use gpu
if the framework is pytorch, it will
from donkeycar.parts.pytorch.torch_train import train
and use it to train.in this train function, it has following codes
however, when creating the trainer, it uses
cpu
flag and never use gpuso I combined them and can use gpus successfully
If it serves no other purpose, I think it's a bug
The text was updated successfully, but these errors were encountered: