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...