Yes, if your model is small enough or, if you are fine-tuning small number of layers. TendorFlow 1.15 and 2.0 are available on Xavier. I understand that PyTorch could be built as well.
Nite that the number of CUDA kernels and amount of memory available is smaller, if compared to descrete Volta GPUs.
You say it can do training for small models because of the presence of the small (512-core) GPU? (plus maybe some left-over, control calculations by the CPU)
dchichkov|5 years ago
Nite that the number of CUDA kernels and amount of memory available is smaller, if compared to descrete Volta GPUs.
fizixer|5 years ago
fizixer|5 years ago
wmf|5 years ago