- Notifications
You must be signed in to change notification settings - Fork26.1k
Install Dependencies
jhelsby edited this pageAug 30, 2024 ·2 revisions
Once you havecloned the source code, enter the PyTorch directory in your terminal and run the following:
conda install cmake ninjapip install -r requirements.txt
Now proceed to the section for your OS below:Linux,MacOS, orWindows.
conda install mkl mkl-include# CUDA only: Add LAPACK support for the GPU if neededconda install -c pytorch magma-cuda110# or the magma-cuda* that matches your CUDA version from https://anaconda.org/pytorch/repo# (optional) If using torch.compile with inductor/triton, install the matching version of triton# Run from the pytorch directory after cloningmake triton
# Add this package on intel x86 processor machines onlyconda install mkl mkl-include# Add these packages if torch.distributed is neededconda install pkg-config libuv
conda install mkl mkl-include# Add these packages if torch.distributed is needed.# Distributed package support on Windows is a prototype feature and is subject to changes.conda install -c conda-forge libuv=1.39
I would love to contribute to PyTorch!