码迷,mamicode.com
首页 > 系统相关 > 详细

CentOS 7搭建Linux GPU服务器

时间:2017-10-04 15:50:07      阅读:254      评论:0      收藏:0      [点我收藏+]

标签:download   export   linux   led   直接   down   logs   png   root   

1. CUDA Toolkit的安装

到https://developer.nvidia.com/cuda-gpus查询GPU支持的CUDA版本:

技术分享

到https://developer.nvidia.com/cuda-downloads,根据操作系统选择下载相应的CUDA Toolkit版本,下载的是一个.run文件,下载完成后以root用户直接运行该文件安装。

安装结束以后。运行:

nvidia-smi

如果列出了GPU状态信息,表明安装成功:

技术分享

2. cuDNN的安装

TensorFlow对神经网络的加速通过cuDNN库实现,所以首先去https://developer.nvidia.com/rdp/cudnn-archive,根据CUDA的版本下载相应版本的cuDNN,也是一个.run文件,下载完成后直接运行。

3. TensorFlow的安装

为了在安装过程中不出现版本冲突等问题,建议先安装Anoconda。到https://www.anaconda.com/download/#linux下载后,运行.sh文件安装。

然后使用下面的命令安装TensorFlow:

conda create -n tensorflow python=2.7
source activate tensorflow
export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc1-cp27-none-linux_x86_64.whl
pip install --ignore-installed --upgrade $TF_BINARY_URL

依次输入:

source activate tensorflow
python

import tensorflow as tf
import pandas as pd
tf.__version__

如果没有报错,则表明安装成功:

技术分享

CentOS 7搭建Linux GPU服务器

标签:download   export   linux   led   直接   down   logs   png   root   

原文地址:http://www.cnblogs.com/mstk/p/7625694.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!