mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
42a4df9447
Add a new option `--cuda` to `tools/nightly.py` to pull the nightly packages with CUDA support. ```bash # installs pytorch-nightly with cpuonly tools/nightly.py pull # The following only available on Linux and Windows # installs pytorch-nightly with latest CUDA we support tools/nightly.py pull --cuda # installs pytorch-nightly with CUDA 12.1 tools/nightly.py pull --cuda 12.1 ``` Also add targets in `Makefile` and instructions in constribution guidelines. ```bash # setup conda environment with pytorch-nightly make setup-env # setup conda environment with pytorch-nightly with CUDA support make setup-env-cuda ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/131133 Approved by: https://github.com/ezyang
56 lines
1.2 KiB
Makefile
56 lines
1.2 KiB
Makefile
# This makefile does nothing but delegating the actual building to cmake.
|
|
PYTHON = python3
|
|
PIP = $(PYTHON) -m pip
|
|
NIGHTLY_TOOL_OPTS := pull
|
|
|
|
all:
|
|
@mkdir -p build && cd build && cmake .. $(shell $(PYTHON) ./scripts/get_python_cmake_flags.py) && $(MAKE)
|
|
|
|
local:
|
|
@./scripts/build_local.sh
|
|
|
|
android:
|
|
@./scripts/build_android.sh
|
|
|
|
ios:
|
|
@./scripts/build_ios.sh
|
|
|
|
clean: # This will remove ALL build folders.
|
|
@rm -r build*/
|
|
|
|
linecount:
|
|
@cloc --read-lang-def=caffe.cloc caffe2 || \
|
|
echo "Cloc is not available on the machine. You can install cloc with " && \
|
|
echo " sudo apt-get install cloc"
|
|
|
|
ensure-branch-clean:
|
|
@if [ -n "$(shell git status --porcelain)" ]; then \
|
|
echo "Please commit or stash all changes before running this script"; \
|
|
exit 1; \
|
|
fi
|
|
|
|
setup-env: ensure-branch-clean
|
|
$(PYTHON) tools/nightly.py $(NIGHTLY_TOOL_OPTS)
|
|
|
|
setup-env-cuda:
|
|
$(MAKE) setup-env PYTHON="$(PYTHON)" NIGHTLY_TOOL_OPTS="$(NIGHTLY_TOOL_OPTS) --cuda"
|
|
|
|
setup_env: setup-env
|
|
setup_env_cuda: setup-env-cuda
|
|
|
|
setup-lint:
|
|
$(PIP) install lintrunner
|
|
lintrunner init
|
|
|
|
setup_lint: setup-lint
|
|
|
|
lint:
|
|
lintrunner
|
|
|
|
quicklint:
|
|
lintrunner
|
|
|
|
triton:
|
|
$(PIP) uninstall -y triton
|
|
@./scripts/install_triton_wheel.sh
|