
本篇主要介绍:在服务器中安装显卡驱动及配置深度学习框架。
服务器与本地主机的区别
在配置本地Linux系统(比如ubuntu)主机的深度学习框架时,显卡驱动的安装按照提示一步一步进行即可。但是,在给服务器安装GPU驱动时,就要考虑到显卡驱动中的显示驱动不能安装,因为服务器预装的系统中没有X桌面(GUI),这点一定要注意。
配置环境
1)安装NVIDIA驱动
显卡驱动下载官网地址:https://www.nvidia.com/Download/index.aspx?lang=en-us
参考下图,CUDA版本一定要选择“8.0”!!!因为TensorFlow官网要求cuda版本为8.0,虽然有写支持9.0,但是实测暂时不支持
新建一个文件夹,存放需要下载的驱动文件
sudo mkdir/home/downloads
点search后点下载,按照下图操作复制链接
链接形式如下:
https://cn.download.nvidia.com/tesla/384.66/nvidia-diag-driver-local-repo-ubuntu1604-384.66_1.0-1_amd64.deb
输入如下命令下载并重命名驱动文件
sudo wget-O driver-384.deb<span style=”color:rgb(0,0,0);”>https://cn.download.nvidia.com/tesla/384.66/nvidia-diag-driver-local-repo-ubuntu1604-384.66_1.0-1_amd64.deb</span>
按照官网“其他”中介绍的安装方式安装
2)安装CUDA toolkit
这里就有小坑了,CUDA版本要在下面“legacy releases”中选择8.0版本
win版本:https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda_9.0.176_win10-exe
Linux ubuntu16.04版本:(可能会变动)https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb
下面这两个用哪个都可以
然后按照下图选择类型,最后按照安装介绍安装。
在安装时,会遇到如下提示,我们不安装显示驱动,所以一定要选“n”
Install NVIDIAAccelerated Graphics Driver for Linux-x86_64 387.26?
(y)es/(n)o/(q)uit:n
最后显示如下内容,代表CUDA安装成功
3)安装CUDNN
在官网下载cudnn-8.0-linux-x64-v5.1.tgz https://developer.nvidia.com/rdp/cudnn-download
需要注册登录才可以下载,下载完同样上传到自己home下面,解压:
tar-xvzf cudnn-8.0-linux-x64-v5.1.tgz
解压后是一个cuda文件夹
cd cuda
sudo cp include/cudnn.h/usr/local/cuda/include
sudo cp lib64/libcudnn.*/usr/local/cuda/lib64
sudo vi/etc/profile
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
source/etc/profile使更改生效。
nvcc-V检查CUDA
下面4)5)步可以参考TensorFlow官网安装教程:
https://www.tensorflow.org/install/install_linux#installing_with_anaconda
4)安装anaconda
5)安装TensorFlow
相关推荐: 链库与湖南云冷1号冷库正式达成战略合作链库与湖南云冷1号冷库正式达成战略合作
3月29日消息,基于对冷链产业互联网发展趋势的共识,湖南云冷1号冷库与链库正式达成战略合作,采用专业化、标准化、系统化的管理模式,借助互联网、大数据等现代化信息技术,为客户提供高时效、多区域的冷链服务,提高企业的市场竞争力。 (图源:“链库网”公众号,下同) …
码刀科技(www.lekshop.cn)是国内知名企业级电商平台提供商,为企业级商家提供最佳的电商平台搭建(多种模式电商平台搭建:B2B/B2B2C/B2C/O2O/新零售/跨境等)、平台管理系统开发及互联网采购解决方案服务, 联系客服了解更多.