1. 程式人生 > >CentOS 7搭建Linux GPU服務器

CentOS 7搭建Linux GPU服務器

系統 查詢 x86 smi bin run 服務器 版本沖突 anaconda

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服務器