环境: 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
在文件中添加以下内容:
vga16fb
nouveau
rivafb
nvidiafb
rivatv
(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