近些年，人工智能的發展迅速，相應的機器學習軟件包也得到了很廣泛的應用，但是由于發展太快，與Android旗下APP編寫軟件類似（Android Studio），存在很多問題，尤其是linux下。本文旨在說明安裝tensorflow的gpu版本，親測有效。

目錄

1.確定tensorflow與CUDA版本的對應關系

2、確定GPU顯卡驅動、cuda之間的對應關系

?3、GPU顯卡驅動安裝

3.1關閉圖形界面

3.2首先禁用nouveau開源驅動

3.3重建initramfs image

3.4重新啟動

3.5查看系統內核版本

3.6安裝CUDA驅動

3.7檢查驅動安裝是否成功

3.8安裝CUDA

3.9安裝CUDNN

3.10打開圖形界面

4、安裝Anaconda

5、安裝Tensorflow

參考資料：

---

?

---

1.確定tensorflow與CUDA版本的對應關系

首先要確定tensorflow與cuda的對應關系，網友提供了下面的對應關系

來自官網：https://tensorflow.google.cn/install/source.

2、確定GPU顯卡驅動、cuda之間的對應關系

確定了要安裝的tensorflow版本以及cuda版本后，需要進一步確定GPU顯卡驅動、cuda之間的對應關系，可以去官網查看相應的網卡對應的驅動，下載下來：

NVIDIA官網鏈接：https://developer.nvidia.com/rdp/cudnn-archive#a-collapse742-10

3、GPU顯卡驅動安裝

3.1關閉圖形界面

systemctl set-default multi-user.target

3.2首先禁用nouveau開源驅動

修改/etc/modprobe.d/blacklist.conf 文件：如果存在blacklist.conf，添加blacklist nouveau，注釋掉blacklist nvidiafb；如果不存在，執行： echo -e "blacklist nouveau\noptions nouveau modeset=0" > /etc/modprobe.d/blacklist.conf

3.3重建initramfs image

cp /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak dracut /boot/initramfs-$(uname -r).img $(uname -r) # 這一步可不執行 rm /boot/initramfs-$(uname -r).img.bak ;

3.4重新啟動

# 改為終端模式，runlevel改為3 systemctl set-default multi-user.target init 3 # 重啟服務器 reboot # sudo shutdown -r now # 重啟后驗證驅動是否被禁用 如果無結果顯示則表明成功禁用 lsmod | grep nouveau

3.5查看系統內核版本

# 不同操作系統的內核版本會不一樣 uname -r # 3.10.0-957.el7.x86_64 # 確認boot目錄的空間不少于300MB df

3.6安裝CUDA驅動

下載相應的驅動run文件，進行安裝，命令如下

# 必需指定kernel source path，否則會報錯；kernel的版本和系統內核有關，可能會有差別 ./NVIDIA-Linux-x86_64-418.87.00.run --kernel-source-path=/usr/src/kernels/3.10.0-957.el7.x86_64 -k $(uname -r)

3.7檢查驅動安裝是否成功

# 查看GPU相關配置 nvidia-smi # 如果正常顯示安裝的顯卡信息，則說明驅動安裝成功； # 如果提示找不到該指令，或什么信息都沒有顯示，則驅動安裝失敗，可以卸載驅動后重新安裝

3.8安裝CUDA

下載相對應版本的cuda版本，進行安裝，主頁上有相應的命令。

3.9安裝CUDNN

下載cuda版本對應的cudnn壓縮包https://developer.nvidia.com/cudnn

解壓cudnn并安裝

$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ $ sudo cp -a cuda/lib64/libcudnn* /usr/local/cuda/lib64/ $ sudo chmod a+r /usr/local/cuda/include/cudnn.h $ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

3.10打開圖形界面

systemctl set-default graphical.target

4、安裝Anaconda

下載Anaconda：https://www.anaconda.com/download/#linux

安裝Anaconda：

$ sudo bash Anaconda3-4.2.0-Linux-x86_64.sh

添加環境變量：

$ gedit ./.bashrc

查看版本檢驗是否安裝完成：

conda --version python --version pip --version

5、安裝Tensorflow

在Anaconda創建python環境：

$ conda create -n py35 python=3.5

激活環境：

$ source activate py35

選擇tensorflow版本，最好使用國內鏡像下載安裝

(py35) $ pip install \ -i https://pypi.tuna.tsinghua.edu.cn/simple/ \ [https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/tensorflow_gpu-1.3.0-cp35-cp35m-linux_x86_64.whl](https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/tensorflow_gpu-1.3.0-cp35-cp35m-linux_x86_64.whl)

檢驗是否安裝成功：

(py35) $ python >>> import tensorflow #退出環境： >>> quit() $ source deactivate py35

安裝keras，sklearn等庫

$ source activate py35 (py35) $ pip install keras -U --pre (py35) $ pip install sklearn

剩下的就是安裝一些編譯器，像Pycharm，Spyder等，這些容易安裝，可以查看相應的資料。

參考資料：

https://www.cnblogs.com/harrymore/p/10307769.html

https://www.jianshu.com/p/c600a89da5da

https://www.cnblogs.com/YSPXIZHEN/p/11466145.html

https://www.jianshu.com/p/98028ef0c52e?from=singlemessage

https://blog.csdn.net/caicaiatnbu/article/details/87626491

https://blog.csdn.net/u012485366/article/details/89981114

?

總結

以上是生活随笔為你收集整理的Centos7.0安装tensorflow的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。