深度學習的搭建

2021-09-10 09:33:20 字數 2558 閱讀 2209

深度學習環境的搭建

所需要安裝的軟體包括要安裝五個軟體ubuntu、cuda、cudnn、tensorflow、pytorch、顯示卡驅動

nvidia: 顯示卡驅動位址

1.安裝顯示卡驅動

(1)安裝驅動之前要解除安裝之前的驅動

開啟終端,首先解除安裝一下之前安裝的驅動

$ sudo apt-get --purge remove nvidia-*

修改配置檔案

修改blacklist配置檔案:

$ sudo gedit /etc/modprobe.d/blacklist.conf

在最後新增幾行:

blacklist vga16fb

blacklist nouveau

blacklist rivafb

blacklist nvidiafb

blacklist rivatv

儲存退出。

禁用nouveau驅動

$ lsmod | grep nouveauud

oapt

inst

alll

ight

dm在執

行sudo apt install lightdm 在執行

sudoap

tins

tall

ligh

tdm在

執行(3)驗證驅動是否成功安裝

$ nvidia-smi

輸入之後就會列印顯示卡驅動的資訊

2.安裝cuda9.0

(1)>>sudo sh cuda_9.0.176_384.81_linux.run

會跳出一堆類似「使用者須知」的東西,直接按ctrl+c跳過後會出現一行字元問你是否accept,輸入accept後回車就可以了

之後會出現一行接一行的字元讓你選擇yes或者no,如下圖,其中第一項問你「install nvidia accelerated graphics driver…?」這裡一定要選擇n,這是要安裝顯示卡驅動我們之前裝過,這裡不用裝,其他選擇y就可以

(3)驗證驅動版本

$ cat /proc/driver/nvidia/version

$ nvcc -v

以上兩條命令執行會有對應的版本號列印出來

3.安裝cudnn

解壓:tar -zxvf cudnn-9.0-linux-x64-v7.tgz

得到如下列表:

cuda/include/cudnn.h

cuda/nvidia_sla_cudnn_support.txt

cuda/lib64/libcudnn.so

cuda/lib64/libcudnn.so.7

cuda/lib64/libcudnn.so.7.0.4

cuda/lib64/libcudnn_static.a

將 include、lib64複製到cuda目錄下:

$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include/

$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/

切換到到本地cuda路徑,建立軟連線,終端輸入:

$ cd /usr/local/cuda/lib64

$ sudo chmod +r libcudnn.so.7.0.4

$ sudo ln -sf libcudnn.so.7.0.4 libcudnn.so.7

$ sudo ln -sf libcudnn.so.7 libcudnn.so

$ sudo ldconfig

查詢cudnn的版本

4.安裝tensorflow

(1)python的處理

ubuntu16.04自帶python3.5和python2.7,如果在命令窗裡輸入python,就會進入python2.7,如果在命令窗裡輸入python3或者python3.5,就會進入python3.5。我覺得python3.5就可以了,就沒有公升級python3.6。你要是想用某乙個python版本,要確保輸入python3後能進入到相應的版本。

(2)接下來我們安裝pip,更明確的說是pip3。輸入指令:

sudo apt-get install python3-pip

然後輸入pip3 -v檢視一下版本,一定要檢查確認該pip從屬於python3.5,也就是你想要用的那個python版本,否則後面安裝的一些庫都是別的python版本的,你要用的python版本就呼叫不了了。

然後在輸入:

pip3 install tensorflow-gpu

等待安裝完畢,之後可能提醒你許可權不夠,就在命令後面加上–user再執行一遍就可以了。

如果你想安裝指定版本的tensorflow,比如1.10.0版的,可以這麼寫:

pip3 install tensorflow-gpu==1.10.0

嘗試呼叫tensorflow,輸入:

$python3

import tensorflow as tf

tf.version

能正確列印tensorflow的版本號,大功告成。

安裝以及報錯參考:

深度學習caffe搭建

經過乙個月的時間,從開始接觸深度學習到環境搭建終於成功了!發表第一篇部落格 慶祝一下 按照21天實戰caffe 進行架構搭建出現的錯誤大致的解決辦法,ubuntu用的是14.04 按照裡面第三天的要求進行操作 後來到第五天發現這些依賴包沒什麼用 直接到第5天 protobuffer 出現錯誤 caf...

深度學習環境搭建

python3.6版本安裝 tensorflow 1.5可以指定安裝版本,目前tensorflow更新到1.7,centos建議安裝1.5版本,其餘版本你的電腦可能不支援 pip3 install upgrade i tensorflow 1.5 pip3 install keras pip3 in...

深度學習環境搭建

windows10 64位 深度學習環境搭建 檢視已建立的環境 conda env list 建立虛擬環境 conda create n name python 3.7 啟用虛擬環境 conda activate name 退出虛擬環境 conda deactivate 刪除虛擬環境 conda r...