一、安装cuda
1.在nvidia官网下载CUDA (linux版本cuda8.0)下载.run文件 下载好后的cuda_8.0.61_375.26_linux.run安装文件移动至当前用户根目录下 终端执行:
2.在安装界面中,可参照下面的描述进行选择
3.添加环境变量,安装结束后,在管理员用户根目录下,找到.bashrc文件并打开,在最后添加下面三行文本,保存退出即可
4.查看duda是否安装好
重启终端后,执行nvcc -V,若显示以下信息,则安装cuda成功
二、安装tensorflow
1. 安装bcupti-dev library可执行
sudo apt-get install libcupti-dev
2.用python直接安装
首先需要pip,如果没有的话可以使用下面的命令安装
sudo apt-get install python-pip python-dev # for Python 2.7
sudo apt-get install python3-pip python3-dev # for Python 3.n
下面使用pip安装tensorflow
$ sudo pip install tensorflow # Python 2.7; CPU support (no GPU support)
$ sudo pip3 install tensorflow # Python 3.n; CPU support (no GPU support)
$ sudo pip install tensorflow-gpu # Python 2.7; GPU support
$ sudo pip3 install tensorflow-gpu # Python 3.n; GPU support
如: sudo pip install tensorflow==1.8.0
三、测试tensorflow是否安装成功
经过一二两个步骤,应该是安装成功了,可写个tensorflow的小程序测试一下
四 、卸载tensorflow
$ pip uninstall tensorflow # Also add -gpu if you installed the GPU version