move spack & update pytorch (#2532)

docs/index.md

@@ -31,6 +31,7 @@ Installation guides
 * [PyTorch for ROCm](./install/pytorch-install.md)
 * [TensorFlow for ROCm](./install/tensorflow-install.md)
 * [MAGMA for ROCm](./install/magma-install.md)
+* [ROCm & Spack](./install/spack-intro.md)
 
 :::
 
@@ -64,7 +65,6 @@ Task-oriented walkthroughs
 * [Setting up for deep learning with ROCm](./how-to/deep-learning-rocm.md)
 * [GPU-enabled MPI](./how-to/gpu-enabled-mpi.md)
 * [System level debugging](./how-to/system-debugging.md)
-* [ROCm & Spack](./how-to/spack-intro.md)
 * [GitHub examples](https://github.com/amd/rocm-examples)
 
 :::

docs/install/pytorch-install.md

@@ -48,8 +48,8 @@ For hardware, software, and third-party framework compatibility between ROCm and
 ## Using a wheels package
 
 PyTorch supports the ROCm platform by providing tested wheels packages. To access this feature, go
-to [https://pytorch.org/get-started/locally/](https://pytorch.org/get-started/locally/). In the interactive
-table, choose ROCm from the _Compute Platform_ row.
+to [https://pytorch.org/get-started/locally/](https://pytorch.org/get-started/locally/). For the correct
+wheels command, you must select 'Linux', 'Python', 'pip', and 'ROCm' in the matrix.
 
 1. Choose one of the following three options:
 
@@ -105,7 +105,8 @@ table, choose ROCm from the _Compute Platform_ row.
    pip3 install wheel setuptools
    ```
 
-3. Install `torch`, `torchvision`, and `torchaudio`, as specified in the [installation matrix](https://pytorch.org/get-started/locally/).
+3. Install `torch`, `torchvision`, and `torchaudio`, as specified in the
+   [installation matrix](https://pytorch.org/get-started/locally/).
 
    :::{note}
    The following command uses the ROCm 5.6 PyTorch wheel. If you want a different version of ROCm,
@@ -136,11 +137,11 @@ table, choose ROCm from the _Compute Platform_ row.
    After installing ROCm PyTorch wheels, run the following code:
 
    ```bash
-   #Optional; replace 'gfx90a' with your architecture
+   #Optional; replace 'gfx90a' with your architecture and 5.6 with your preferred ROCm version
    export GFX_ARCH=gfx90a
 
    #Optional
-   export ROCM_VERSION=5.5
+   export ROCM_VERSION=5.6
 
    ./install_kdb_files_for_pytorch_wheels.sh
    ```

docs/sphinx/_toc.yml.in

@@ -35,6 +35,8 @@ subtrees:
     title: Tensorflow for ROCm
   - file: install/magma-install.md
     title: MAGMA for ROCm
+  - file: how-to/spack-intro.md
+    title: ROCm & Spack
 
 - caption: Compatibility & support
   entries:
@@ -80,8 +82,6 @@ subtrees:
         title: MI200
       - file: how-to/tuning-guides/w6000-v620.md
         title: RDNA2
-  - file: how-to/spack-intro.md
-    title: ROCm & Spack
   - url: https://github.com/amd/rocm-examples
     title: GitHub examples