Ubuntu18.04 安装 TensorFlow (GPU)

重新在台式机上学习深度学习,Ubuntu18.04 + Tensorflow-gpu + cuda8.0 + cuDNN6.0

一、准备工作(有一些不是必需的):

1.安装pip:

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

2.安装vim:

sudo apt-get install vim-gtk

在命令行下,输入命令:sudo vim /etc/vim/vimrc    必须加上sudo,否则你是没有权限编辑vimrc的。

在这个文件中,会有这么一句: syntax on   意思是语法高亮,如果您的被注释掉了,请“让它出来”。

请在您的VIM的最后一行,输入他们,可以让您的VIM变得更漂亮、舒服。

set nu                           // 在左侧行号

set tabstop                  //tab 长度设置为 4

set nobackup               //覆盖文件时不备份

set cursorline               //突出显示当前行

set ruler                       //在右下角显示光标位置的状态行

set autoindent             //自动缩进

然后 ESC :wq保存即可


二、ubuntu安装GPU显卡驱动

打开终端:sudo apt-get update

选择系统设置→软件更新→附加驱动→选择nvidia最新驱动→应用更改.(如果没有,去官网找对应的驱动安装即可)

验证安装成功:软件连找到 NVIDIA X Sever Settings


驱动安装成功!


三、安装Tensorflow依赖的编译工具bazel

1、安装bazel前,需先安装JDK8

sudo apt-get install software-properties-common

sudo add-apt-repository ppa:webupd8team/java

sudo apt-get update

sudo apt-get install oracle-java8-installer

安装完成后 验证java版本:java -version


2、安装bazel

echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list

sudo apt install curl

安装和更新Bazel

sudo apt-get update && sudo apt-get install bazel

sudo apt-get upgrade bazel 


三、安装cuda 8.0(先看本章 4、验证cuda8是否安装成功)

1、cuda 8.0 下载地址:https://developer.nvidia.com/cuda-80-ga2-download-archive


验证主机是否满足cuda8.0安装的要求

  • 3
    点赞
  • 39
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值