首頁 > 軟體

NVIDIA CuDNN 安裝說明-用於機器學習的GPU庫

2020-06-16 17:09:07

安裝cudnn-7.0-Linux-x64-v4.0-rc.tgz

去百度網路硬碟下載原始檔 連結: https://pan.baidu.com/s/1slWaAVb 密碼: 5aua

NVIDIA cuDNN是用於深度神經網路的GPU加速庫。它強調效能、易用性和低記憶體開銷。NVIDIA cuDNN可以整合到更高階別的機器學習框架中,如加州大學伯克利分校的流行CAFFE軟體。簡單的,插入式設計可以讓開發人員專注於設計和實現神經網路模型,而不是調整效能,同時還可以在GPU上實現高效能現代平行計算。

主要特點

· 專為NVIDIA GPU特調的,用於折積神經網路向前和向後的折積程式。

· 專為最新的NVIDIA GPU架構優化

· 針對4緯張量的任意維度排序,striding和次區域可以很容易整合到任何神經網路的執行中

· 對於許多其他常見布局型別(ReLU, Sigmoid, Tanh, pooling, softmax )向前和向後的路徑

· 基於上下文的API,可以很容易地多執行緒

安裝:

sudo tar xvf cudnn-7.0-linux-x64-v4.0-rc.tgz

cd cuda/include

sudo cp *.h /usr/local/include/

cd ../lib64

sudo cp lib* /usr/local/lib/

cd /usr/local/lib

sudo chmod +r libcudnn.so.4.0.4

sudo ln -sf libcudnn.so.4.0.4 libcudnn.so.4

sudo ln -sf libcudnn.so.4 libcudnn.so

sudo ldconfig

Ubuntu 14.04 安裝設定CUDA  http://www.linuxidc.com/Linux/2014-10/107501.htm

Ubuntu 14.04下CUDA8.0 + cuDNN v5 + Caffe  安裝設定  http://www.linuxidc.com/Linux/2017-01/139300.htm

Caffe設定簡明教學 ( Ubuntu 14.04 / CUDA 7.5 / cuDNN 5.1 / OpenCV 3.1 ) http://www.linuxidc.com/Linux/2016-09/135016.htm

Ubuntu 16.04 安裝設定MATLAB+Python +CUDA8.0+cuDNN+OpenCV3.1的Caffe環境  http://www.linuxidc.com/Linux/2017-06/145087.htm

在Ubuntu 14.04上設定CUDA+Caffe+cuDNN+Anaconda+DIGITS  http://www.linuxidc.com/Linux/2016-11/136775.htm

本文永久更新連結地址http://www.linuxidc.com/Linux/2017-08/146393.htm


IT145.com E-mail:sddin#qq.com