一. Pip安裝:
$ sudo apt-get install python-pip python-dev # for python
$ sudo apt-get install python3-pip python3-dev # for python3
二.安裝英偉達顯卡驅動
1查看自身版本
lspci | grep -i vga
lspci | grep -i nvidia
或去官網上查看適合你GPU的驅動
http://www.nvidia.com/Download/index.aspx?lang=en-us
2安裝
對ubuntu16 可通過 系統設置-軟件和更新-附加驅動 安裝
3
執行完上述後,重啓reboot
安裝完後再運行
nvidia-smi 來看看 如果出現了你的GPU列表,則說明驅動安裝成功了。 另外也可以通過,或者輸入nvidia-settings 查看
三.安裝cuda
https://developer.nvidia.com/cuda-downloads據其所說方式安裝
ubuntu的gcc編譯器是5.4.0,然而cuda8.0不支持5.0以上的編譯器,因此需要降級,把編譯器版本降到4.9:
在terminal中執行
sudo apt-get install gcc-4.9 gcc-5 g++-4.9 g++-5
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10
sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30
sudo update-alternatives --set cc /usr/bin/gcc
sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30
sudo update-alternatives --set c++ /usr/bin/g++
配置cuda8.0之後主要加上的一個環境變量聲明,
gedit ~/.bashrc
在文件~/.bashrc末尾加上
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
通過命令查看CUDA安裝是否成功
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
四.cudnn
https://developer.nvidia.com/cudnn tensorflow1.1 支持cudnnv5.1
cd ~/Downloads/
tar zxvf cudnn-8.0-linux-x64-v5.1.tgz #解壓文件
# 複製頭文件和lib文件
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/
# 進入文件夾
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5 #刪除原有動態文件
# 然後修改文件權限,並創建新的軟連接
sudo chmod u=rwx,g=rx,o=rx libcudnn.so.5.1.10 #若這裏不修改權限會有如下錯誤
sudo ln -s libcudnn.so.5.1.5 libcudnn.so.5 #生成軟銜接
sudo ln -s libcudnn.so.5 libcudnn.so #生成軟鏈接
# 錯誤
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
ImportError: No module named rensorflow
>>> import tensorflow
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/home/yojiu/.local/lib/python2.7/site-packages/tensorflow/__init__.py", line 24, in <module>
from tensorflow.python import *
File "/home/yojiu/.local/lib/python2.7/site-packages/tensorflow/python/__init__.py", line 51, in <module>
from tensorflow.python import pywrap_tensorflow
File "/home/yojiu/.local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 52, in <module>
raise ImportError(msg)
ImportError: Traceback (most recent call last):
File "/home/yojiu/.local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 41, in <module>
from tensorflow.python.pywrap_tensorflow_internal import *
File "/home/yojiu/.local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>
_pywrap_tensorflow_internal = swig_import_helper()
File "/home/yojiu/.local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper
_mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
ImportError: libcudnn.so.5: cannot open shared object file: No such file or directory
Failed to load the native TensorFlow runtime.
See https://www.tensorflow.org/install/install_sources#common_installation_problems
for some common reasons and solutions. Include the entire stack trace
above this error message when asking for help.
五.安裝tensorflow
直接配置URL在線 whl 進行安裝:
$ sudo pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.1.0rc2-cp27-none-linux_x86_64.whl
#或其他網站鏡像