Ubuntu16.04+CUDA8.0+cuDNN5.1+Python2.7+TensorFlow1.2.0环境搭建

软件版本说明:我选的Linux系统是Ubuntu16.04,CUDA用的8.0Ubuntu16.04+CUDA8.0+cuDNN5.1+Python2.7只支持TensorFlow1.3.0如下的版本,因此个人TensorFlow版本选择为1.2.0。若是想使用TensorFlow1.3.0,就须要CUDA8.0+cuDNN5.1了。html

软件下载地址:CUDAhttps://developer.nvidia.com/cuda-toolkit-archivepython

cuDNNhttps://developer.nvidia.com/cudnnlinux


1.Ubuntu16.04安装TensorFlowCPU版本:ubuntu

(1)首先安装pipvim

  打开终端,输入命令:sudo apt-get install python-pip python-devui

(2)而后安装tensorflow:spa

  pip install tensorflowcode


2.Ubuntu16.04安装TensorFlowCPU版本:orm

 (1)安装NVIDIA显卡驱(提早下载好): htm

  安装NVIDIA显卡驱,首先须要禁用自带的 nouveau nvidia驱动:sudo vim /etc/modprobe.d/blacklist-nouveau.conf,而且在文件中添加:

    blacklist nouveau
    options nouveau modeset=0

  而后再更新一次:sudo update-initramfs -u  ,修改后须要重启系统。重启后,确认下Nouveau已经被禁用,使用命令: lsmod | grep nouveau
  同时须要结束x-window的服务,不然驱动将没法正常安装,命令:sudo service lightdm stop

  而后切换到终端:Ctrl+Alt+F1

  安装驱动:



























  1. cd /home/用户名/   2. sudo sh ./NVIDIA*.run  而后重启X-Windowsudo service lightdm startCtrl+Alt+F7进入图形界面,检查驱动是否安装成功:nvidia-smi (2)gcc降级:ubuntugcc编译器是5.4.0,由于cuda8.0不支持5.0以上的编译器,因此须要把编译器版本降到4.9(3)安装CUDA在终端依次输入:cd /home/用户名/cuda安装包路径下sudo dpkg -i cuda-repo-ubuntu1604-8-0-rc_8.0.27-1_amd64​.debsudo apt-get updatesudo apt-get install cuda(4)安装CUDNN:在终端依次输入:cd /home/用户名/Desktop/cudnn安装包路径下 tar xvzf cudnn-8.0-linux-x64-v5.1-ga.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*这里,安装完成以后须要配置环境变量。(5)安装其余库:Tensorflow官网上,根据提示安装。(6)安装Bazel依赖:由于是使用tensorflow源码编译/安装,因此须要使用 bazel buildhttps://www.bazel.io/versions/master/docs/install.html按照Bazel官网提供的说明进行安装,安装完成以后须要配置环境变量。(7)安装tensorflowGPU版):pip install tensorflow-gpu
相关文章
相关标签/搜索