Note: This section is only required for installing
the Conda package. If you are using the Docker image, you can skip this section.
- GPU-related Software
- Install GPU related software according to the operating system. For Ubuntu 16.04, install CUDA 9.0, cuDNND 7 and Driver 384 or above. For Ubuntu 18.04, install CUDA 10.0, cuDNN 7 and Driver 410 or above.
- NVIDIA GPU Drivers
- Install GPU driver by apt-get or directly install the CUDA package with
driver. For
example:
apt-get install nvidia-384 apt-get install nvidia-410
- CUDA Toolkit
- Get the CUDA package associated with the Ubuntu version from https://developer.nvidia.com/cuda-toolkit-archive and directly install the NVIDIA CUDA runfile package.
- cuDNN SDK
- Get cuDNN from https://developer.nvidia.com/cudnn and append its installation directory to
the $LD_LIBRARY_PATH environmental
variable.
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cudnn-7.0.5/lib64
- CUPTI
- CUPTI is required by
vai_p_tensorflow
and is installed together with CUDA. You must add the CUPTI directory to the $LD_LIBRARY_PATH environment variable. For example:export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64
- NCCL
- NCCL is required by
vai_p_caffe
. Download NCCL from its homepage (https://developer.nvidia.com/nccl/nccl-legacy-downloads) and install.sudo dpkg -i nccl-repo-ubuntu1804-2.6.4-ga-cuda10.0_1-1_amd64.deb sudo apt update sudo apt install libnccl2=2.6.4-1+cuda10.0 libnccl-dev=2.6.4-1+cuda10.0