# neuralODE not using multiple CPUs

**URL:** <https://discourse.julialang.org/t/neuralode-not-using-multiple-cpus/66873>\
**Category:** Machine Learning\
**Tags:** question\
**Created:** [August 23, 2021, 8:09pm UTC](https://discourse.julialang.org/t/neuralode-not-using-multiple-cpus/66873 "2021-08-23T20:09:53Z")\
**Posts on this page:** 4\
**Page:** 1

<div class="post-metadata">

**Author:** ![DiracFermion1](https://avatars.discourse-cdn.com/v4/letter/d/2acd7d/32.png) [@DiracFermion1](https://discourse.julialang.org/u/DiracFermion1)\
**Post date:** [August 23, 2021, 8:09pm UTC](https://discourse.julialang.org/t/neuralode-not-using-multiple-cpus/66873/1 "2021-08-23T20:09:53Z")

</div>

Hi, I am trying to run neuralODE with multiple CPUs, but I found that when I set `export JULIA_NUM_THREADS=10`, and `Threads.nthreads()` outputs `10`, the neuralODE model is still using only 1 CPU.

I am wondering if anyone has had similar issue before and what may be a good way to fix it?

Thank you!

---

<div class="post-metadata">

**Author:** ![ChrisRackauckas](https://sea2.discourse-cdn.com/julialang/user_avatar/discourse.julialang.org/chrisrackauckas/32/77_2.png) [@ChrisRackauckas](https://discourse.julialang.org/u/ChrisRackauckas)\
**Post date:** [August 24, 2021, 12:49pm UTC](https://discourse.julialang.org/t/neuralode-not-using-multiple-cpus/66873/2 "2021-08-24T12:49:09Z")

</div>

What are you running? Are you using big matuls with MKL.jl? Did you follow the tutorial [https://diffeqflux.sciml.ai/dev/examples/data\_parallel/](https://diffeqflux.sciml.ai/dev/examples/data_parallel/) ?

---

<div class="post-metadata">

**Author:** ![DiracFermion1](https://avatars.discourse-cdn.com/v4/letter/d/2acd7d/32.png) [@DiracFermion1](https://discourse.julialang.org/u/DiracFermion1)\
**Post date:** [August 24, 2021, 2:13pm UTC](https://discourse.julialang.org/t/neuralode-not-using-multiple-cpus/66873/4 "2021-08-24T14:13:20Z")

</div>

I am currently using `NeuralODE` class, it looks like the link you provided is mainly about `ODEProblem` , I am wondering if there are any differences or which would be better to use for parallelization?

Thank you!

---

<div class="post-metadata">

**Author:** ![ChrisRackauckas](https://sea2.discourse-cdn.com/julialang/user_avatar/discourse.julialang.org/chrisrackauckas/32/77_2.png) [@ChrisRackauckas](https://discourse.julialang.org/u/ChrisRackauckas)\
**Post date:** [August 24, 2021, 2:22pm UTC](https://discourse.julialang.org/t/neuralode-not-using-multiple-cpus/66873/5 "2021-08-24T14:22:40Z")

</div>

ODEProblem is more general, NeuralODE is specifically for neural network only ODEs. In NeuralODE, parallelization is done by putting the batches into different columns.