溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》

CentOS 7搭建Linux GPU服務(wù)器的教程

發(fā)布時(shí)間:2020-10-15 21:56:53 來(lái)源:腳本之家 閱讀:172 作者:MSTK 欄目:服務(wù)器

CentOS 7搭建Linux GPU服務(wù)器的步驟,供大家參考,具體內(nèi)容如下

1. CUDA Toolkit的安裝

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

CentOS 7搭建Linux GPU服務(wù)器的教程

到https://developer.nvidia.com/cuda-downloads,根據(jù)操作系統(tǒng)選擇下載相應(yīng)的CUDA Toolkit版本,下載的是一個(gè).run文件,下載完成后以root用戶直接運(yùn)行該文件安裝。

安裝結(jié)束以后。運(yùn)行:

復(fù)制代碼 代碼如下:
nvidia-smi

如果列出了GPU狀態(tài)信息,表明安裝成功:

CentOS 7搭建Linux GPU服務(wù)器的教程

2. cuDNN的安裝

TensorFlow對(duì)神經(jīng)網(wǎng)絡(luò)的加速通過(guò)cuDNN庫(kù)實(shí)現(xiàn),所以首先去https://developer.nvidia.com/rdp/cudnn-archive,根據(jù)CUDA的版本下載相應(yīng)版本的cuDNN,也是一個(gè).run文件,下載完成后直接運(yùn)行。

3. TensorFlow的安裝

為了在安裝過(guò)程中不出現(xiàn)版本沖突等問(wèn)題,建議先安裝Anoconda。到https://www.anaconda.com/download/#linux下載后,運(yùn)行.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__

如果沒(méi)有報(bào)錯(cuò),則表明安裝成功:

CentOS 7搭建Linux GPU服務(wù)器的教程

以上就是本文的全部?jī)?nèi)容,希望對(duì)大家的學(xué)習(xí)有所幫助,也希望大家多多支持億速云。

向AI問(wèn)一下細(xì)節(jié)

免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱:is@yisu.com進(jìn)行舉報(bào),并提供相關(guān)證據(jù),一經(jīng)查實(shí),將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI