新聞中心
CUDA(Compute Unified Device Architecture,簡稱CUDA)是由英偉達(dá)推出的用于GPU(Graphic Processing Unit,圖形處理器)的并行計算的程序開發(fā)接口。它有助于減少開發(fā)時間,讓硬件加速應(yīng)用程序開發(fā)人員能夠更加輕松地實現(xiàn)GPU加速。而CUDNN(CUDA Deep Neural Network)是由英偉達(dá)公司提供的深度神經(jīng)網(wǎng)絡(luò)的加速代碼庫,使用方便實用。

為了在linux中使用CUDNN,我們首先要安裝CUDA,并在 systemd 中啟用cuda 自啟動服務(wù)。具體操作:
1. 下載CUDA環(huán)境,下載地址:https://developer.nvidia.com/cuda-downloads
2. 安裝CUDA:在終端中輸入 sudo sh cuda**.run –override 并運行就可以完成安裝。
3. 自啟動CUDA服務(wù):系統(tǒng)啟動的時候,會自動檢測CUDA環(huán)境,去讓它自啟動我們可以通過systemd去實現(xiàn),輸入 sudo systemctl enable cuda 即可。
接下來,我們可以開始安裝CUDNN,可以在官網(wǎng)下載需要版本的CUDNN:https://developer.nvidia.com/cudnn
1. 下載CUDNN:官網(wǎng)上下載,在終端輸入wget ,下載后運行 sudo sh cudnn**.run
2. 安裝CUDNN:將CUDNN安裝在CUDA根目錄,輸入 sudo sh cudnn**.run –override –silent
安裝完成之后,我們還需要為我們的環(huán)境配置CUDNN庫:
1. 配置環(huán)境變量:輸入sudo vi /etc/environment 新建:
CUDNN_HOME=”/usr/local/cuda/lib64/”
LD_LIBRARY_PATH=”$LD_LIBRARY_PATH:$CUDNN_HOME”
既添加成功,將系統(tǒng)變量設(shè)置生效:source /etc/environment
2. 將lib文件拷貝到/usr/local/cuda/lib64:輸入 sudo cp lib* /usr/local/cuda/lib64
4. 將include文件拷貝到/usr/local/cuda/include:輸入 sudo cp include /usr/local/cuda/include
經(jīng)過以上步驟,我們就可以在Linux上運行CUDNN,并實現(xiàn)通過GPU加速深度學(xué)習(xí)計算。CUDNN加速深度學(xué)習(xí),解決了可移植性和通用性問題,提供簡便的開發(fā)和運行環(huán)境,擬能讓機器學(xué)習(xí)技術(shù)應(yīng)用更加普及化。
創(chuàng)新互聯(lián)成都網(wǎng)站建設(shè)公司提供專業(yè)的建站服務(wù),為您量身定制,歡迎來電(028-86922220)為您打造專屬于企業(yè)本身的網(wǎng)絡(luò)品牌形象。
成都創(chuàng)新互聯(lián)品牌官網(wǎng)提供專業(yè)的網(wǎng)站建設(shè)、設(shè)計、制作等服務(wù),是一家以網(wǎng)站建設(shè)為主要業(yè)務(wù)的公司,在網(wǎng)站建設(shè)、設(shè)計和制作領(lǐng)域具有豐富的經(jīng)驗。
網(wǎng)站題目:Linux下運行CUDNN的技術(shù)實現(xiàn)(linuxcudnn)
文章地址:http://m.fisionsoft.com.cn/article/cdsihgi.html


咨詢
建站咨詢
