Hello, I had tried to train custom data set using CPU, but i got message like this
Anyone, can help me to solve it?
i had tried add --device cpu when running the program
/site-packages/torch/amp/autocast_mode.py:198: UserWarning: User provided device_type of 'cuda', but CUDA is not available. Disabling
warnings.warn('User provided device_type of \'cuda\', but CUDA is not available. Disabling')
Killed
Hello, I had tried to train custom data set using CPU, but i got message like this Anyone, can help me to solve it?
i had tried add --device cpu when running the program
/site-packages/torch/amp/autocast_mode.py:198: UserWarning: User provided device_type of 'cuda', but CUDA is not available. Disabling warnings.warn('User provided device_type of \'cuda\', but CUDA is not available. Disabling') Killed