-
Horovod 모듈 설치하기Practical_skills/odds and ends 2020. 1. 16. 20:19
refer
refer
cuda 는 알아서 설치 요망
prerequisit: NCCL for Multi-GPU, gcc compllier, anaconda gxx complier, tensorflow-gpu, openmpi- install nccl
- install gcc version 4.9
- if you use anaconda virtual environment, install anaconda g++ complier
- install tensorflow-gpu
- install openmpi
- install horovod with tensorflow
install nccl
refer
Download Network Installer for Ubuntu16.04$ sudo dpkg -i nvidia-machine-learning-repo-<version>.deb $ sudo apt update $ sudo apt install libnccl2=2.5.6-1+cuda10.0 libnccl-dev=2.5.6-1+cuda10.0
version = 2.5.6
version checklocate nccl| grep "libnccl.so" | tail -n1 | sed -r 's/^.*\.so\.//'
install gcc version 4.9
GCC, G++ 버전관리
refersudo update-alternatives --display gcc sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 40 --slave /usr/bin/g++ g++ /usr/bin/g++-7 sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 20 --slave /usr/bin/g++ g++ /usr/bin/g++-5 sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 60 --slave /usr/bin/g++ g++ /usr/bin/g++-4.9 gcc --version
if you use anaconda virtual environment, install anaconda g++ complier
conda install -c anaconda gxx\_linux-64
install tensorflow-gpu
conda install -c anaconda tensorflow-gpu=2.0.0
install openmpi
conda install -c conda-forge openmpi
install horovod with tensorflow
NCCL 헤더와 라이브러리 파일 경로, cuda 경로를 잘 넣어주어야 한다.HOROVOD\_NCCL\_INCLUDE=/usr/include HOROVOD\_NCCL\_LIB=/usr/lib/x86\_64-linux-gnu HOROVOD\_CUDA\_HOME=/usr/local/cuda-10.0 HOROVOD\_WITH\_TENSORFLOW=1 pip install --force-reinstall --no-deps --no-cache-dir horovod==0.18.1
'Practical_skills > odds and ends' 카테고리의 다른 글
apt-get 으로 설치가 안될 때 PPA Repository 추가하기 (0) 2019.08.27 libpcap 을 사용하여 packet capture하기 (0) 2019.08.26 sublime text를 사용하여 c++ 개발환경 설정하기 (1) 2019.08.26 Google map, chart 사용하기 (개요) (0) 2019.07.14