Not able to train Model on CPU mode

Robotics & Edge Computing Jetson Systems Jetson TX1
2

Hi,

Thanks for your question.
The error indicates Caffe tried to use cuDNN but it’s not in GPU mode.

Architecture is not an issue for training a model.
Digits saves model with general caffe format and is nothing related to GPU architecture.

Please feel free to train a model on desktop GPU and then port it to TX1.
In TX1, tensorRT will compile the model into tensorRT format based on GPU architecture.