TensorFlow环境搭建及安装教程

时间:2021-11-14 12:50:37

1、安装虚拟环境virtualenv相关配置(创建了python3.5的环境)

TensorFlow环境搭建及安装教程


2、http://docs.nvidia.com/cuda/cuda-installation-guide-linux/#axzz4VZnqTJ2A,这个⽹址去
检查相关配置系统是否满⾜CUDA要求,之后去官⽹下载cuda*.deb⽹络版本,这⾥⾯有驱动程
序,还有别的本地下载⽅式可以根据选择下载
如图:

TensorFlow环境搭建及安装教程

TensorFlow环境搭建及安装教程

TensorFlow环境搭建及安装教程


然后就可以等很久了,如果前期检查不通过的话,这⼀步会有问题


3、去.bashrc配置CUDA相关环境

TensorFlow环境搭建及安装教程

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

TensorFlow环境搭建及安装教程

 


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


2、程序所需构建⼯具安装

TensorFlow环境搭建及安装教程

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