ABSTRACT
descripe : tensorflow installation guide
date : 2018.06.06
platform : NVIDIA GeForce 1080Ti
version : ubuntu - 16.04 LTS
NVIDIA-driver - 384.13
CUDA - 9.0
cuDNN - 7.1.2
tensorflow - 1.8
==========================
STEP I : install nvidia driver
reference : https://blog.csdn.net/u012759136/article/details/53355781
1. sudo apt-get remove –purge nvidia*
2. sudo vim /etc/modprobe.d/blacklist.conf
3. add "blacklist nouveau" at the end of file
4. sudo update-initramfs -u
5. sudo reboot
6. Ctrl+Alt+F1
7. lsmod | grep nouveau
8. sudo /etc/init.d/lightdm stop
9. cd ${nvidia_driver_path}
10. sudo ./NVIDIA-Linux-x86_64-384.13.run
11. sudo reboot
Please run the commands, if the desktop can not login after reboot.
1. cd /usr/share/X11/xorg.conf.d/
2. sudo mv nvidia-drm-outputclass.conf nvidia-drm-outputclass.conf.bak
3. sudo reboot
==========================
STEP II : install CUDA
1. Download CUDA 9.0 at https://developer.nvidia.com/cuda-90-download-archive
select Linux, x86_64, Ubuntu, 16.04, runfile(local) and download base installer
2. cd ${cuda_9.0_path}
3. sudo ./cuda_9.0.176_384.81_linux.run
4. enter n, y, '\n', y, y, '\n' step by step, like this. ('\n' means enter)
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81?
(y)es/(n)o/(q)uit: n
Install the CUDA 9.0 Toolkit?
(y)es/(n)o/(q)uit: y
Enter Toolkit Location
[ default is /usr/local/cuda-9.0 ]:
Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: y
Install the CUDA 9.0 Samples?
(y)es/(n)o/(q)uit: y
Enter CUDA Samples Location
[ default is /home/ai_group ]:
Installing the CUDA Toolkit in /usr/local/cuda-9.0 ...
Missing recommended library: libGLU.so
Missing recommended library: libX11.so
Missing recommended library: libXi.so
Missing recommended library: libXmu.so
Missing recommended library: libGL.so
Installing the CUDA Samples in /home/ai_group ...
Copying samples to /home/ai_group/NVIDIA_CUDA-9.0_Samples now...
Finished copying samples.
===========
= Summary =
===========
Driver: Not Selected
Toolkit: Installed in /usr/local/cuda-9.0
Samples: Installed in /home/ai_group, but missing recommended libraries
5. gedit ./.bashrc
6. add environment path at the end
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
==========================
STEP III : install cuDNN
1. Download cuDNN for CUDA 9.0 at https://developer.nvidia.com/rdp/cudnn-archive
2. Extract the cuDNN
3. cd ${cuDNN_extract_folder_path}
4. sudo cp ./include/cudnn.h /usr/local/cuda/include
5. sudo cp ./lib64/libcudnn* /usr/local/cuda/lib64
6. sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
==========================
STEP IV : install tensorflow
1. choose the ailiyun as the default source
2. sudo apt-get update
3. sudo apt-get python3-pip
4. sudo pip3 install --upgrade pip
5. pip3 install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple/
to see other way of installation, visit https://tensorflow.google.cn/install/install_linux
==========================
STEP V : test
1. creat an empty document at desktop named "test.py" and enter the code showed below.
#!/usr/bin/env python3
import tensorflow as tf
hello = tf.constant('Hello,world!')
sess = tf.Session() # creat a conversation
result = sess.run(hello) # run
sess.close() # close
print(result) # show result
2. cd ~/Desktop
3. python3 test.py
Congratulations, if “Hello,world!” is printed at the end. Enjoy!