Skip to content

Instantly share code, notes, and snippets.

@gau-nernst

gau-nernst/setup.sh

Created September 6, 2025 04:56
Show Gist options
  • Save gau-nernst/1a00df9d5f5bf6f49e6b4cdcba08effb to your computer and use it in GitHub Desktop.
Save gau-nernst/1a00df9d5f5bf6f49e6b4cdcba08effb to your computer and use it in GitHub Desktop.
Download ZIP
Set up PyTorch dev env
Raw
setup.sh
uv venv --python=3.12 --managed-python
source .venv/bin/activate
uv pip install --group dev
uv pip install ninja # or sudo apt install ninja-build
USE_DISTRIBUTED=0 USE_MKLDNN=0 BUILD_TEST=0 USE_FBGEMM=0 USE_NNPACK=0 USE_QNNPACK=0 USE_XNNPACK=0 USE_FLASH_ATTENTION=0 USE_MEM_EFF_ATTENTION=0 uv pip install --no-build-isolation -v -e .
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment