Skip to content

Instantly share code, notes, and snippets.

View rfeinman's full-sized avatar

Reuben Feinman rfeinman

View GitHub Profile
@rfeinman
rfeinman / build_pytorch_openmp_mkl_macOS.sh
Last active March 20, 2022 16:15
Build PyTorch with MKL & OpenMP threading on macOS
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
# This install used the following development environment:
# - We have installed xcode command line tools
# - We have installed cmake via homebrew (my version is 3.20.2)
# - We have created a fresh conda environment "pytorch-dev" and installed
# python=3.8 and all of the packages from pytorch/README.md except cmake
# (including the torch.distributed deps)
# - We have copied the relevant OpenMP header files into a fresh directory at ~/openmp
# - We have cloned the pytorch github repo to ~/pytorch
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~