Xuehai Pan 42a4df9447 Support CUDA nightly package in tools/nightly.py (#131133) ...

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

2024-07-25 05:33:52 +00:00

..

install-dev-tools.sh

Support CUDA nightly package in tools/nightly.py (#131133)

2024-07-25 05:33:52 +00:00

update_alternatives_clang.sh

Add devcontainer support to PyTorch Project (#98252)

2023-05-30 19:44:18 +00:00