TensorFlow環境搭建及安裝教程


1、安裝虛擬環境virtualenv相關配置(創建了python3.5的環境)


2、http://docs.nvidia.com/cuda/cuda-installation-guide-linux/#axzz4VZnqTJ2A,這個⽹址去
檢查相關配置系統是否滿⾜CUDA要求,之后去官⽹下載cuda*.deb⽹絡版本,這⾥⾯有驅動程
序,還有別的本地下載⽅式可以根據選擇下載
如圖:


然后就可以等很久了,如果前期檢查不通過的話,這⼀步會有問題


3、去.bashrc配置CUDA相關環境

source .bashrc


4、檢查是否安裝成功

$ cd /usr/local/cuda-8.0/samples/5_Simulations/nbody
$ sudo make
$ ./nbody

If successful, a new window will popup running n-body simulation.


5、下載cuDNN:cudnn-8.0-linux-x64-v5.1.tgz

然后運⾏以下命令

$ tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*


6、在3.5版本的環境中安裝TensorFlow(我們使⽤的是1.0.1版本)

pip install tensorflow-gpu==1.0.1 --trusted-host pypi.douban.com

 


找到⼀個非常清楚地官⽅教程⽹址:https://www.nvidia.com/en-us/data-center/gpuaccelerated-applications/tensorflow/


2、程序所需構建⼯具安裝

https://docs.bazel.build/versions/master/install-ubuntu.html


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM