This wiki lists the steps for setting up a developer environment to contribute code changes to PyTorch.

To develop PyTorch you will need:

• Python 3.8 or later (for Linux, Python 3.8.1+ is needed)
• A C++17 compatible compiler, such as clang

If you want to compile with CUDA support, install the following (note that CUDA is not supported on macOS)

• NVIDIA CUDA 11.0 or above
• NVIDIA cuDNN v8 or above
• Compiler compatible with CUDA

Note: You could refer to thecuDNN Support Matrix for cuDNN versions with the various supported CUDA, CUDA driver and NVIDIA hardware

If you want to compile with ROCm support, install

• AMD ROCm 4.0 and above installation
• ROCm is currently supported only for Linux systems.

• Fork, clone, and checkout the PyTorch source