首页 技术 正文
技术 2022年11月6日
0 收藏 820 点赞 945 浏览 2096 个字

目录

本次使用主机配置:

cpu:i7-8700K     显卡:GTX-1080Ti

一、安装显卡驱动

打开命令窗口(ctrl+Alt+T)

sudo apt-get purge nvidia*sudo add-apt-repository ppa:graphics-drivers/ppasudo apt-get update sudo apt-get install nvidia- nvidia-settings

如果报错add-apt-repository不存在,运行以下代码解决:

sudo apt-get install software-properties-common python-software-properties

检验是否安装成功

nvidia-smi

显示以下结果表明安装成功

Ubuntu 16 server 安装 tensorflow-GPU

二、安装CUDA

由于目前NVIDIA官网上的CUDA9.0+版本 存在缺陷,本次安装选择CUDA8.0

下载安装包:

地址:https://developer.nvidia.com/cuda-80-ga2-download-archive

Ubuntu 16 server 安装 tensorflow-GPU

Ubuntu 16 server 安装 tensorflow-GPU

进入下载页面后,下载CUDA8.0(cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb)安装包和

一个升级包(cuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64.deb)

安装步骤:

1、安装基础包

sudo dpkg -i cuda-repo-ubuntu1604---local-ga2_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda 

2、安装升级包

sudo dpkg -i cuda-repo-ubuntu1604---local-cublas-performance-update_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get upgrade cuda

设置环境变量:

在.bashrc中添加

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda

查看安装版本

cat /usr/local/cuda/version.txt

输出

CUDA Version 8.0.

三、安装cuDNN

cuDNN6.0和CUDA8.0最佳拍档,所以本次选择cuDNN6.0

下载安装包:

地址:https://developer.nvidia.com/rdp/cudnn-download

Ubuntu 16 server 安装 tensorflow-GPU

安装:

1、将下载的压缩包复制到CUDA目录

sudo cp cudnn-8.0-linux-x64-v6..tgz /usr/local/cuda

2、进入/usr/local/cuda解压文件,然后建立连接

tar -zxvf cudnn-8.0-linux-x64-v6..tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d

查看cuDNN版本信息:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 

输出

#define CUDNN_MAJOR      6
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 21
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)#include "driver_types.h"

检测CUDA和cuDNN是否安装成功:

进入测试目录

//进入测试目录:
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery//编译环境
make -j4//运行:
./deviceQuery

结果中有Result = PASS 表示安装成功

Ubuntu 16 server 安装 tensorflow-GPU

四、安装tensorflow-gpu

//安装命令,要安装哪个版本的tensorflow-gpu,使用 ‘tensorflow-gpu==xx’就行了
pip install tensorflow-gpu==1.4//卸载命令
pip uninstall tensorflow-gpu

测试gpu是否安装成功

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

输出里有 GPU的名称,显存等信息,表示tensorflow可以使用GPU了

Ubuntu 16 server 安装 tensorflow-GPU

到此安装已经完成,如有问题,欢迎留言讨论。

相关推荐
python开发_常用的python模块及安装方法
adodb:我们领导推荐的数据库连接组件bsddb3:BerkeleyDB的连接组件Cheetah-1.0:我比较喜欢这个版本的cheeta…
日期:2022-11-24 点赞:878 阅读:8,996
Educational Codeforces Round 11 C. Hard Process 二分
C. Hard Process题目连接:http://www.codeforces.com/contest/660/problem/CDes…
日期:2022-11-24 点赞:807 阅读:5,510
下载Ubuntn 17.04 内核源代码
zengkefu@server1:/usr/src$ uname -aLinux server1 4.10.0-19-generic #21…
日期:2022-11-24 点赞:569 阅读:6,353
可用Active Desktop Calendar V7.86 注册码序列号
可用Active Desktop Calendar V7.86 注册码序列号Name: www.greendown.cn Code: &nb…
日期:2022-11-24 点赞:733 阅读:6,137
Android调用系统相机、自定义相机、处理大图片
Android调用系统相机和自定义相机实例本博文主要是介绍了android上使用相机进行拍照并显示的两种方式,并且由于涉及到要把拍到的照片显…
日期:2022-11-24 点赞:512 阅读:7,770
Struts的使用
一、Struts2的获取  Struts的官方网站为:http://struts.apache.org/  下载完Struts2的jar包,…
日期:2022-11-24 点赞:671 阅读:4,848