@@ -55,19 +55,19 @@ source functorch-env/bin/activate
55
55
Next, install one of the following following PyTorch nightly binaries.
56
56
```
57
57
# For CUDA 10.2
58
- pip install --pre torch -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html
58
+ pip install --pre torch -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html --upgrade
59
59
# For CUDA 11.1
60
- pip install --pre torch -f https://download.pytorch.org/whl/nightly/cu111/torch_nightly.html
60
+ pip install --pre torch -f https://download.pytorch.org/whl/nightly/cu111/torch_nightly.html --upgrade
61
61
# For CPU-only build
62
- pip install --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
62
+ pip install --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html --upgrade
63
63
```
64
64
If you already have a nightly of PyTorch installed and wanted to upgrade it
65
65
(recommended!), append ` --upgrade ` to one of those commands.
66
66
67
67
Install functorch:
68
68
```
69
69
pip install ninja # Makes the build go faster
70
- pip install --user "git+https://github.com/facebookresearch /functorch.git"
70
+ pip install --user "git+https://github.com/pytorch /functorch.git"
71
71
```
72
72
73
73
Run a quick sanity check in python:
@@ -292,7 +292,7 @@ If you use functorch in your publication, please cite it by using the following
292
292
@Misc{functorch2021,
293
293
author = {Horace He, Richard Zou},
294
294
title = {functorch: JAX-like composable function transforms for PyTorch},
295
- howpublished = {\url{https://github.com/facebookresearch /functorch}},
295
+ howpublished = {\url{https://github.com/pytorch /functorch}},
296
296
year = {2021}
297
297
}
298
298
```
0 commit comments