frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Removing conda references from PyTorch Docs (#152702)

Browse Source 
Addresses #148339

Pull Request resolved: https://github.com/pytorch/pytorch/pull/152702
Approved by: https://github.com/svekars, https://github.com/albanD, https://github.com/atalman
This commit is contained in:
Anita Katahoire
2025-05-20 20:33:28 +00:00
committed by PyTorch MergeBot
parent 05bc78e64f
commit 996c4d803d
6 changed files with 18 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
functorch/dim/README.md
View File

@ -51,19 +51,18 @@ We may eventually upstream them into PyTorch itself along with `functorch`.
We have to install a nightly build of PyTorch so first set up an environment:
```sh
conda create --name dim
conda activate dim
python -m venv dim
source dim/bin/activate # or `& .\dim\Scripts\Activate.ps1` on Windows
```
First-class dims requires a fairly recent nightly build of PyTorch so that functorch will work. You can install it using one of these commands:
```sh
# For CUDA 10.2
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch-nightly
# For CUDA 11.3
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch-nightly
# For CUDA
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu116
# For CPU-only build
conda install pytorch torchvision torchaudio cpuonly -c pytorch-nightly
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
```
Install dim. You will be asked for github credentials to access the fairinternal organization.