一、安装NVIDA驱动
安装驱动大致有两种方法:
- 直接上NVIDA官网找到适合本机显卡型号的驱动,下载安装
- 利用ubuntu系统本身自带的显卡驱动
在这里演示最简单的方法2
2、在“附加驱动”里,系统会自动搜索N卡驱动,列表里会提供对应你显卡的最新版官方驱动
3、最后点“应用更改”,等待安装完毕。
4、重启系统即可
二、安装Anaconda
1、https://www.anaconda.com/download/ 下载
2、cd到文件下载的位置输入如下命令安装Anaconda
sudo sh Anaconda3-4.4.0-Linux-x86_64.sh
3、安装
安装过程一路按默认安装,输入“yes‘
当出现提示信息“Do you wish to proceed with the installation of Microsoft VSCode? [yes|no]”,输入no
4、安装完毕后加入环境变量
编辑文件
sudo gedit ~/.bashrc
在文件末尾加入实际安装路径(这个路径仅供参考)
export PATH="/home/wangdong/anaconda3/bin:$PATH"
立即生效
source ~/.bashrc
三、安装tensorflow-gpu
1、创建一个环境,指定python版本
conda create -n tensorflow-gpu python=3.5
source activate tensorflow-gpu
conda install tensorflow-gpu