(一)安装tensorflow

来源:互联网 发布:电视软件论坛 编辑:程序博客网 时间:2024/05/21 11:16

环境: ubuntu16.04 GTX1050

1. 安装cuda8.0

下载地址:https://developer.nvidia.com/cuda-downloads

(1)按 ctrl+alt+F1 进入tty, 登录tty后输入如下命令

sudo service lightdm stop

该命令会关闭lightdm。

(2)屏蔽集显

sudo vim /etc/modprobe.d/blacklist.conf

在文件中添加以下内容:

vga16fbnouveaurivafbnvidiafbrivatv

(3) 删除以前的nvidia驱动

sudo apt-get remove –purge nvidia*

(4)下载合适的驱动
http://www.geforce.cn/drivers
(5)运行cuda 安装文件进行安装

2. 安装tensorflow

参考网址:https://www.tensorflow.org/install/install_linux#the_url_of_the_tensorflow_python_package
tensorflow有多种安装方式,这里选择virtualenv安装
(1)安装所有必备工具:

sudo apt-get install python-pip python-dev python-virtualenv # for Python 2.7

(2) 建立一个全新的 virtualenv 环境

virtualenv --system-site-packages targetDirectory # for Python 2.7

(3)激活 virtualenv

source ~/tensorflow/bin/activate # bash, sh, ksh, or zsh

(4) virtualenv 内, 安装 TensorFlow:

 (tensorflow)$ pip install --upgrade tfBinaryURL   # Python 2.7 地址在https://www.tensorflow.org/install/install_linux#the_url_of_the_tensorflow_python_package