博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
问题记录 | 配置ubuntu18.04+cuda9.0+cudnn服务器tensorflow-gpu深度学习环境
阅读量:5135 次
发布时间:2019-06-13

本文共 1574 字,大约阅读时间需要 5 分钟。

因为实验室服务器资源有限,我被分配的服务器经常变化,但是常常就分到连显卡驱动以及cuda都没有装的服务器,真的很头疼,我已经配了四五台了,特此记录一下,以便以后直接照版本安装。

Install nvidia driver

我的服务器配置是ubuntu18.04,显卡是1080Ti。

给ubuntu添加驱动的源:

sudo add-apt-repository ppa:graphics-drivers/ppasudo apt updatesudo apt upgrade

查看设备型号得到推荐安装的驱动型号:

ubuntu-drivers devices

自动安装显卡驱动

sudo ubuntu-drivers autoinstall

重启

sudo reboot

如果重启一切顺利就命令检查

lsmod | grep nvidia

or

nvidia-smi

Install CUDA9.0 and cudnn7

查看 CUDA 版本:

cat /usr/local/cuda/version.txt

查看 CUDNN 版本:

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

下载好cuda_9.0.176_384.81_linux.run

链接:

ubuntu16.04版本的:

安装cuda:

sudo chmod 777 cuda_9.0.176_384.81_linux.runsudo ./cuda_9.0.176_384.81_linux.run -toolkit -samples -override --no-opengl-libs -silent

更改gcc版本

sudo apt install gcc-5 g++-5sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 50 sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 50

设置cuda环境变量

vim ~/.bashrc1
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:/usr/local/cuda/extras/CPUTI/lib64export CUDA_HOME=/usr/local/cuda-9.0/binexport PATH=$PATH:$LD_LIBRARY_PATH:$CUDA_HOME

下载cudnn要在这个https://developer.nvidia.com/rdp/cudnn-archive链接中选择下载

复制cudnn库至cuda中

tar -zxvf cudnn-9.0-linux-x64-v7.tgzsudo cp cuda/lib64/* /usr/local/cuda-9.0/lib64/sudo cp cuda/include/* /usr/local/cuda-9.0/include/

安装tensoflow-gpu-1.14:

pip install -U tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple --user

tensorflow测试程序:

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

转载于:https://www.cnblogs.com/ManWingloeng/p/11489122.html

你可能感兴趣的文章
[Swift]LeetCode922.按奇偶排序数组 II | Sort Array By Parity II
查看>>
php match_model的简单使用
查看>>
SIP服务器性能测试工具SIPp使用指导(转)
查看>>
Vue_(组件通讯)子组件向父组件传值
查看>>
STM32单片机使用注意事项
查看>>
032. asp.netWeb用户控件之一初识用户控件并为其自定义属性
查看>>
移动开发平台-应用之星app制作教程
查看>>
leetcode 459. 重复的子字符串(Repeated Substring Pattern)
查看>>
springboot No Identifier specified for entity的解决办法
查看>>
浅谈 unix, linux, ios, android 区别和联系
查看>>
51nod 1428 活动安排问题 (贪心+优先队列)
查看>>
latex for wordpress(一)
查看>>
如何在maven工程中加载oracle驱动
查看>>
Flask 系列之 SQLAlchemy
查看>>
aboutMe
查看>>
【Debug】IAR在线调试时报错,Warning: Stack pointer is setup to incorrect alignmentStack,芯片使用STM32F103ZET6...
查看>>
一句话说清分布式锁,进程锁,线程锁
查看>>
FastDFS使用
查看>>
服务器解析请求的基本原理
查看>>
[HDU3683 Gomoku]
查看>>