caffe安裝中元件cuda和cudnn配置方法

2021-08-15 21:19:47 字數 1233 閱讀 4228

首先解除安裝已經安裝的cuda:

sudo yum remove /usr/local/cuda-7.5
檢視資訊1:

lspci | grep -i nvidia
檢視資訊2:

uname -m && cat /etc/*release
檢視資訊3:

gcc --version
檢視資訊4:

uname -r
查詢cuda版本:

yum search cuda
yum install cuda-8-0
修改環境變數:

vim ~/.bashrc
在最後加入以下幾行:

export cuda_home=/usr/local/cuda-8.0  

export path=/usr/local/cuda-8.0/bin:$path

export ld_library_path=/usr/local/cuda-8.0/lib64:$ld_library_path

export ld_library_path="/usr/local/cuda-8.0/lib:$"

修改好之後檢視:

nvcc -v

nvidia-smi

tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz

cp -p /usr/local/cuda/include/cudnn.h /usr/local/cuda-8.0/include

cp -p /usr/local/cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64

chmod a+r /usr/local/cuda-8.0/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn*

ps:關於cuda和顯示卡的問題

訓練過程中發現問題:

沒有執行任何的程式 使用cuda7.5 發現但是gpu利用率為100%。

解決辦法:

需要把驅動模式設定為常駐記憶體才可以,設定命令:

nvidia-smi -pm 1

docker安裝和caffe安裝

第一步 啟動ubuntu安裝docker root lyf virtual machine apt get install docker.io 第二步 第三步 開啟docker root lyf virtual machine sudo service docker start root lyf v...

妙算cuda 和 opencv 安裝

refference 原文裝了兩邊opencv,已更正 2.執行以下命令安裝 cuda repo l4t r21.3 6 5 prod 6.5 42 armhf.deb 至作業系統。sudo dpkg i cuda repo l4t r21.3 6 5 prod 6.5 42 armhf deb s...

CUDA和CUDNN的安裝

2.cudnn 驗證選擇相應版本後 linux x86 64 ubuntu 18.04 runfile local 給.run檔案賦執行許可權 chmod x cuda run sudo cuda run閱讀說明可以直接ctrl c跳過 第二個選項安裝驅動選no 之前已經安裝好了 其餘全部accep...