【发布时间】:2020-06-19 16:14:13
【问题描述】:
我正在尝试在 Tensorflow 概率中使用 MCMC 进行采样,但它似乎只使用了一个 CPU 内核。是否可以并行使用多个 CPU?
【问题讨论】:
-
我自己没有找到比手动将链数设置为低值并自行启动许多并行运行更好的方法 - 不过我也很好奇,如果有人知道的话!
标签: tensorflow mcmc tensorflow-probability
我正在尝试在 Tensorflow 概率中使用 MCMC 进行采样,但它似乎只使用了一个 CPU 内核。是否可以并行使用多个 CPU?
【问题讨论】:
标签: tensorflow mcmc tensorflow-probability
您可以尝试使用这两个参数: set_intra_op_parallelism_threads 和 set_inter_op_parallelism_threads.
但是,据说自动选择了:
0 表示系统选择了一个合适的数字。
并且设置大的数字并不能保证所有节点都会被使用,也不保证它们会被使用到 100%。
【讨论】: