Skip to content

Remove OMP_NUM_THREADS and MKL_NUM_THREADS settings from docker images#12836

Closed
yf225 wants to merge 1 commit intopytorch:masterfrom
yf225:pytorch_docker_image_253
Closed

Remove OMP_NUM_THREADS and MKL_NUM_THREADS settings from docker images#12836
yf225 wants to merge 1 commit intopytorch:masterfrom
yf225:pytorch_docker_image_253

Conversation

@yf225
Copy link
Copy Markdown
Contributor

@yf225 yf225 commented Oct 18, 2018

OMP_NUM_THREADS and MKL_NUM_THREADS are set to 4 by default in the docker images, which causes nproc to only show 4 cores in the docker containers by default, and building PyTorch is slow in this default case. We likely don't need these two flags to be set, and this PR tests that hypothesis.

@yf225 yf225 changed the title [WIP] Remove OMP_NUM_THREADS and MKL_NUM_THREADS settings from docker images Remove OMP_NUM_THREADS and MKL_NUM_THREADS settings from docker images Oct 19, 2018
@yf225
Copy link
Copy Markdown
Contributor Author

yf225 commented Oct 19, 2018

Corresponding commit in ossci-job-dsl: pytorch/ossci-job-dsl@7444382

Copy link
Copy Markdown
Contributor

@facebook-github-bot facebook-github-bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

yf225 has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

@ezyang ezyang added the merged label Jun 25, 2019
laurentdupin pushed a commit to laurentdupin/pytorch that referenced this pull request Apr 24, 2026
pytorch#12836)

Summary:
`OMP_NUM_THREADS` and `MKL_NUM_THREADS` are set to 4 by default in the docker images, which causes `nproc` to only show 4 cores in the docker containers by default, and building PyTorch is slow in this default case. We likely don't need these two flags to be set, and this PR tests that hypothesis.
Pull Request resolved: pytorch#12836

Differential Revision: D10468218

Pulled By: yf225

fbshipit-source-id: 7a57962c962e162a8d97f730626825aa1e371c7f
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants