Skip to content
Snippets Groups Projects
Verified Commit f88b6fcd authored by Felix Soubelet's avatar Felix Soubelet
Browse files

update as pytorch 1.12 is no longer nightly

parent b8cf98fc
No related branches found
No related tags found
1 merge request!19Apple Silicon page updates
Pipeline #4938157 passed
......@@ -121,11 +121,6 @@ conda activate pytorch
python -m pip install torch torchvision torchaudio
```
As the `1.12` version is currently a nightly build, the impatient can install it with:
```bash
python -m pip install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu
```
??? info "Making PyTorch use the GPU"
The PyTorch integration with Metal is not as seamless as the TensorFlow one, and just like with any other accelerators, PyTorch requires you to explicitly set the `device` for calculations.
To use the Apple Silicon GPU, one has to specify the device as `mps` (Metal Performance Shaders) in either the `torch.device` constructor, or when creating tensors:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment