diff --git a/README.md b/README.md index 772625fee..fe143b68c 100644 --- a/README.md +++ b/README.md @@ -102,8 +102,7 @@ conda update -n monarchenv --all -c conda-forge -y sudo dnf install -y libibverbs rdma-core libmlx5 libibverbs-devel rdma-core-devel # Install build dependencies -pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu128 -pip install -r build-requirements.txt +pip install -r torch-requirements.txt -r build-requirements.txt # Install test dependencies pip install -r python/tests/requirements.txt @@ -142,8 +141,7 @@ export CXX=clang++ sudo apt install -y cuda-toolkit-12-8 cuda-12-8 # Install build dependencies -pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu128 -pip install -r build-requirements.txt +pip install -r torch-requirements.txt -r build-requirements.txt # Install test dependencies pip install -r python/tests/requirements.txt diff --git a/build-requirements.txt b/build-requirements.txt index 88c14d487..11fa6deea 100644 --- a/build-requirements.txt +++ b/build-requirements.txt @@ -1,4 +1,3 @@ -torch setuptools setuptools-rust wheel diff --git a/torch-requirements.txt b/torch-requirements.txt new file mode 100644 index 000000000..8482b8ed7 --- /dev/null +++ b/torch-requirements.txt @@ -0,0 +1,2 @@ +--index-url https://download.pytorch.org/whl/nightly/cu128 +torch --pre