shadimsaleh / install-Tensorflow-on-Drive-PX2

Explain how to install tensorflow on NVIDIA DRIVE PX2

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

install-Tensorflow-on-Drive-PX2

Install tensorflow on NVIDIA DRIVE PX2(AutoChauffeur)
CUDA Version 9.0, CUDNN Version 7
(check cuda version by writing 'cat /usr/local/cuda/version.txt' in terminal)
(check cudnn version by writing 'cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2' in terminal)

Setup Environment

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH

Install pip

  • for python 2.7
    sudo apt-get install python-pip

  • for python 3.6
    sudo apt-get install python3-pip

Install tensorflow 1.9.0

Install tensorflow 1.11.0

My build whl file

you can download in here CUDA 9.0 (tf-1.9, 1.10), 9.2 (tf-1.9) https://drive.google.com/drive/folders/1q9i0yqKaEMFFgSLIZauNHIWT82UD7T4w?usp=sharing

About

Explain how to install tensorflow on NVIDIA DRIVE PX2