added installation instructions for dali

m-tassano · Mar 30, 2020 · d26c8af · d26c8af
1 parent 15d252d
commit d26c8af
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -32,6 +32,10 @@ The code as is runs in Python +3.6 with the following dependencies:
 * [OpenCV](https://pypi.org/project/opencv-python/)
 * [tensorboardX](https://github.com/lanpa/tensorboardX/)
 
+Note: the tested version of NVIDIA DALI is 0.10.0. To install it (supposing you have CUDA 10.0), you need to run
+```
+pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/cuda/10.0 nvidia-dali==0.10.0 
+```
 ## Usage
 
 ### Testing