Skip to content

Commit

Permalink
update all docs for pt19 variant (#2515)
Browse files Browse the repository at this point in the history
Signed-off-by: Joel Polizzi <quic_jpolizzi@quicinc.com>
  • Loading branch information
quic-jpolizzi authored Oct 20, 2023
1 parent 0d29fef commit c4ea588
Show file tree
Hide file tree
Showing 3 changed files with 13 additions and 6 deletions.
3 changes: 3 additions & 0 deletions Docs/install/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,9 @@ PyTorch

#. **torch-gpu** for PyTorch 1.13 GPU package with Python 3.8 and CUDA 11.x - *Recommended for use* with PyTorch models
#. **torch-cpu** for PyTorch 1.13 CPU package with Python 3.8 - If installing on a machine without CUDA
#. **torch-gpu-pt19** for PyTorch 1.9 GPU package with Python 3.8 and CUDA 11.x
#. **torch-cpu-pt19** for PyTorch 1.9 CPU package with Python 3.8 - If installing on a machine without CUDA


TensorFlow

Expand Down
8 changes: 6 additions & 2 deletions Docs/install/install_docker.rst
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,8 @@ Set variant
Set the `<variant_string>` to ONE of the following depending on your desired variant
#. For the PyTorch 1.13 GPU variant, use `torch-gpu`
#. For the PyTorch 1.13 CPU variant, use `torch-cpu`
#. For the PyTorch 1.9 GPU variant, use `torch-gpu-pt19`
#. For the PyTorch 1.9 CPU variant, use `torch-cpu-pt19`
#. For the TensorFlow GPU variant, use `tf-gpu`
#. For the TensorFlow CPU variant, use `tf-cpu`
#. For the ONNX GPU variant, use `onnx-gpu`
Expand Down Expand Up @@ -120,8 +122,10 @@ Go to https://github.com/quic/aimet/releases and identify the release tag of the

Set the <variant_string> to ONE of the following depending on your desired variant

#. For the PyTorch GPU variant, use "torch_gpu"
#. For the PyTorch CPU variant, use "torch_cpu"
#. For the PyTorch 1.13 GPU variant, use "torch_gpu"
#. For the PyTorch 1.13 CPU variant, use "torch_cpu"
#. For the PyTorch 1.9 GPU variant, use "torch_gpu_pt19"
#. For the PyTorch 1.9 CPU variant, use "torch_cpu_pt19"
#. For the TensorFlow GPU variant, use "tf_gpu"
#. For the TensorFlow CPU variant, use "tf_cpu"
#. For the ONNX GPU variant, use "onnx_gpu"
Expand Down
8 changes: 4 additions & 4 deletions Docs/install/install_host.rst
Original file line number Diff line number Diff line change
Expand Up @@ -158,10 +158,10 @@ Go to https://github.com/quic/aimet/releases and identify the release tag of the

Set the <variant_string> to ONE of the following depending on your desired variant

#. For the PyTorch 1.9 GPU variant, use "torch_gpu"
#. For the PyTorch 1.9 CPU variant, use "torch_cpu"
#. For the PyTorch 1.13 GPU variant, use "torch_gpu_pt113"
#. For the PyTorch 1.13 CPU variant, use "torch_cpu_pt113"
#. For the PyTorch 1.13 GPU variant, use "torch_gpu"
#. For the PyTorch 1.13 CPU variant, use "torch_cpu"
#. For the PyTorch 1.9 GPU variant, use "torch_gpu_pt19"
#. For the PyTorch 1.9 CPU variant, use "torch_cpu_pt19"
#. For the TensorFlow GPU variant, use "tf_gpu"
#. For the TensorFlow CPU variant, use "tf_cpu"
#. For the ONNX GPU variant, use "onnx_gpu"
Expand Down

0 comments on commit c4ea588

Please sign in to comment.