From 9a27b916da21a0143eb5497ee7ec6dced69f2358 Mon Sep 17 00:00:00 2001 From: Zhiqiang Wang Date: Mon, 14 Mar 2022 15:46:12 +0800 Subject: [PATCH] Fix building docs and update docstrings (#363) * Add TORCH_PATH to LD_LIBRARY_PATH * Fixing denpendenncies * Add PyYAML>=5.3.1 to requirements --- deployment/libtorch/README.md | 8 ++++---- docs/requirements.txt | 1 + requirements.txt | 1 + 3 files changed, 6 insertions(+), 4 deletions(-) diff --git a/deployment/libtorch/README.md b/deployment/libtorch/README.md index a3ec4ace..e971cd2a 100644 --- a/deployment/libtorch/README.md +++ b/deployment/libtorch/README.md @@ -4,13 +4,12 @@ The LibTorch inference for `yolort`, both GPU and CPU are supported. ## Dependencies -- Ubuntu 18.04 -- LibTorch 1.8.0 / 1.9.0 -- TorchVision 0.9.0 / 0.10.0 +- Ubuntu / Windows / macOS +- LibTorch 1.8.0+ together with corresponding TorchVision 0.9.0+ - OpenCV 3.4+ - CUDA 10.2 \[Optional\] -*We didn't impose too strong restrictions on the version of CUDA and Ubuntu systems.* +*We didn't impose too strong restrictions on the version of CUDA.* ## Usage @@ -18,6 +17,7 @@ The LibTorch inference for `yolort`, both GPU and CPU are supported. ```bash export TORCH_PATH=$(dirname $(python -c "import torch; print(torch.__file__)")) + export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$TORCH_PATH/lib/ # Optional ``` 1. Don't forget to compile `LibTorchVision` using the following scripts. diff --git a/docs/requirements.txt b/docs/requirements.txt index b60d038b..e4955f7f 100644 --- a/docs/requirements.txt +++ b/docs/requirements.txt @@ -4,3 +4,4 @@ ipython sphinx sphinx-material nbsphinx +PyYAML>=5.3.1 diff --git a/requirements.txt b/requirements.txt index 53bd963c..913a19da 100644 --- a/requirements.txt +++ b/requirements.txt @@ -13,6 +13,7 @@ matplotlib>=3.2.2 numpy>=1.18.5 Pillow>=8.0.0 +PyYAML>=5.3.1 scipy>=1.4.1 tqdm>=4.41.0