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!

neuralODE not using multiple CPUs

ChrisRackauckas August 24, 2021, 12:49pm 2

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/ ?

Topic		Replies	Views
GPU simulation time of neural ODE program Machine Learning diffeq , flux	8	916	August 16, 2019
Flux parallel execution Machine Learning flux	3	2788	March 29, 2019
Flux.jl and the state of multi-processing Machine Learning	2	1653	February 27, 2019
How can I make Flux use all my CPUs? General Usage	10	3227	March 15, 2019
Batch evaluation for NeuralODE? Machine Learning	1	1305	August 21, 2021