kimishpatel:

caffe2::pthreadpool()->set_thread_count(1)


@dbalchev sorry for late response. For some reason I did not get any notification. Unfortunately at the moment we dont have python API for this. Possibly in next release, or we can thin of introducing it in master if that may work for you.