Skip to content

Instantly share code, notes, and snippets.

@pbamotra
Created July 3, 2019 21:07
Show Gist options
  • Save pbamotra/4815c076c5a11adf95ca7729e08e005f to your computer and use it in GitHub Desktop.
Save pbamotra/4815c076c5a11adf95ca7729e08e005f to your computer and use it in GitHub Desktop.
DALI Post-1.2
# Find out the cuda version so that we install appropriate DALI binaries
# Find installation instructions at
# https://github.com/NVIDIA/DALI#installing-prebuilt-dali-packages
$ nvcc --version
# sample output
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130
# install Nvidia DALI python bindings
pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/cuda/10.0 nvidia-dali
# sample output
Looking in indexes: https://pypi.org/simple, https://developer.download.nvidia.com/compute/redist/cuda/10.0
Collecting nvidia-dali
Downloading https://developer.download.nvidia.com/compute/redist/cuda/10.0/nvidia-dali/nvidia_dali-0.8.1-699137-cp36-cp36m-manylinux1_x86_64.whl (30.0MB)
100% |████████████████████████████████| 30.0MB 1.3MB/s
Requirement already satisfied: future in /usr/local/lib/python3.6/dist-packages (from nvidia-dali) (0.16.0)
Installing collected packages: nvidia-dali
Successfully installed nvidia-dali-0.8.1
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment