【環(huán)境搭建】ubuntu18.04+CUDA10.1+cuDNN7.6.4+TensorFlow1.15.0

一、WIN10 + ubuntu18.04雙系統(tǒng)安裝

  1. rufus制作系統(tǒng)安裝盤,注意選GPT類型
  2. 關(guān)閉WIN10快速啟動(dòng)以及BIOS的SECURE BOOT選項(xiàng)
  3. BIOS選擇U盤啟動(dòng)項(xiàng),進(jìn)入ubuntu安裝界面
  4. 只需自定義/分區(qū)和HOME分區(qū),注意引導(dǎo)要寫入WINDOWS BOOT MANAGER,方便從開(kāi)機(jī)從UEFI選擇進(jìn)入ubuntu還是win10系統(tǒng),而不是依賴于grub引導(dǎo)
  5. 系統(tǒng)重裝,推薦使用easyUefi,先恢復(fù)WIN10引導(dǎo),然后刪除ubuntu所在分區(qū)卷,這樣比較保險(xiǎn)

二、CUDA10.1安裝

  1. NVIDIA官網(wǎng)測(cè)評(píng)顯卡型號(hào),決定驅(qū)動(dòng)程序版本和CUDA版本
  2. CUDA的安裝推薦使用apt-get方式,會(huì)連同顯卡驅(qū)動(dòng)一并安裝,不推薦先手動(dòng)安裝顯卡驅(qū)動(dòng),后續(xù)如果發(fā)現(xiàn)顯卡驅(qū)動(dòng)版本和CUDA版本不適配,顯卡驅(qū)動(dòng)會(huì)很難卸載干凈,最后只能重裝系統(tǒng)
  3. CUDA安裝完成后,重啟電腦,nvidia-smi命令驗(yàn)證顯卡驅(qū)動(dòng)版本,nvcc -V驗(yàn)證CUDA版本

三、cuDNN配合CUDA版本安裝

  1. 選對(duì)版本,下載壓縮包
  2. 解壓到/usr/local,主要是往CUDA目錄添加庫(kù)文件
  3. 添加環(huán)境變量到~/.bashrc,添加路徑到/etc/ld.so.conf,sudo ldconfig,重啟電腦,這步很關(guān)鍵,TensorFlow容易找不到CUDA庫(kù)文件

四、Anaconda安裝

  • 主要是為了安裝Python虛擬環(huán)境方便,conda create命令創(chuàng)建虛擬環(huán)境,指定Python版本

五、TensorFlow安裝

  • TensorFlow官網(wǎng)教程很詳細(xì),但是下載速度非常慢,建議找國(guó)內(nèi)的鏡像,關(guān)鍵就是匹配前一步的Python版本,指定gpu版以及選對(duì)版本,pip命令安裝即可
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容