Skip to content

Commit 89d5080

Browse files
Jennifer Yangniuxjxlnxjanifer112x
authored andcommitted
3.0 fix (#995)
* update url of replace_pytorch.sh * fix the docker install html link (#994) Co-authored-by: qiuyuny <[email protected]> Co-authored-by: niuxj <[email protected]> Co-authored-by: qiuyuny <[email protected]>
1 parent 523627d commit 89d5080

File tree

3 files changed

+4
-5
lines changed

3 files changed

+4
-5
lines changed

docker/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,4 +5,4 @@
55
</tr>
66
</table>
77

8-
This folder contains the Docker container build files. For information on building and using Vitis AI Docker containers, start [here](https://xilinx.github.io/Vitis-AI/3.0/html/docs/install/install.html)
8+
This folder contains the Docker container build files. For information on building and using Vitis AI Docker containers, start [here](https://xilinx.github.io/Vitis-AI/docs/install/install.html)

docs/docs/install/install.html

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@ <h2>Host / Development Software Installation<a class="headerlink" href="#host-de
165165
<p>There are three primary options for installation:</p>
166166
<ul class="simple">
167167
<li><p>[Option1] Directly leverage pre-built Docker containers available from Docker Hub:
168-
<a class="reference external" href="https://hub.docker.com/r/xilinx/vitis-ai/tags">xilinx/vitis-ai</a>.</p></li>
168+
<a class="reference external" href="https://hub.docker.com/r/xilinx/">xilinx/vitis-ai</a>.</p></li>
169169
<li><p>[Option2] Build a custom container to target your local host machine.</p></li>
170170
<li><p>[Option3] Install Vitis AI on AWS or Azure. See the instructions to install Vitis AI on <a class="reference internal" href="install_on_aws.html"><span class="doc">AWS</span></a>, or <a class="reference internal" href="install_on_azure.html"><span class="doc">Azure</span></a>.</p></li>
171171
</ul>
@@ -494,6 +494,5 @@ <h4>Building the Container<a class="headerlink" href="#building-the-container" t
494494
});
495495
</script>
496496

497-
498497
</body>
499-
</html>
498+
</html>

src/vai_quantizer/vai_q_pytorch/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ vai_q_pytorch is designed as a part of a standard platform for neural network de
2828

2929
If you work in Vitis-AI 3.0 version of docker, there is a conda environment "vitis-ai-pytorch", in which vai_q_pytorch package is already installed.
3030
In this conda environment, python version is 3.7, pytorch version is 1.12 and torchvision version is 0.13. You can directly start our "resnet18" example without installation steps.
31-
A new Conda environment with a specified PyTorch version (1.2~1.12) can be created using the script [replace_pytorch.sh](https://github.com/Xilinx/Vitis-AI/blob/master/docker/dockerfiles/replace_pytorch.sh). This script clones a Conda environment from vitis-ai-pytorch, uninstalls the original PyTorch, Torchvision and vai_q_pytorch
31+
A new Conda environment with a specified PyTorch version (1.2~1.12) can be created using the script [replace_pytorch.sh](https://github.com/Xilinx/Vitis-AI/blob/master/docker/common/replace_pytorch.sh). This script clones a Conda environment from vitis-ai-pytorch, uninstalls the original PyTorch, Torchvision and vai_q_pytorch
3232
packages, and then installs the specified version of PyTorch, Torchvision, and re-installs vai_q_pytorch from source code.
3333
- Copy example/resnet18_quant.py to docker environment
3434
- Download pre-trained [Resnet18 model](https://download.pytorch.org/models/resnet18-5c106cde.pth)

0 commit comments

Comments
 (0)