ubuntu配置深度學(xué)習(xí)環(huán)境

image

目前最新的 Ubuntu Server 16.04.3 LTS 系統(tǒng)為例,介紹包括 CUDA、CudNN 軟件以及 TensorFlow、Keras 等 Python 庫的安裝方法。

下面是需要安裝的軟件列表。

  • Cuda
  • CudNN
  • TensorFlow, Pytorch
  • OpenCV,Caffe
  • Matlab

前期準(zhǔn)備: sudo apt-get update, 找到清華大學(xué)鏡像站,下好anaconda并安裝,然后將anaconda的python添加到環(huán)境變量

然后下載到NVIDIA下載cuda

image

一個坑: 必須<meta charset="utf-8">關(guān)閉圖形界面,不然會報(bào)錯

sudo service lightdm stop

image

cuda安裝好

image

下面安裝cuDNN, 用來加速深度學(xué)習(xí)框架

到英偉達(dá)官網(wǎng)下載好壓縮包,然后解壓縮。將cuda/lib64的文件復(fù)制到 /usr/local/cuda/lib64中,cuda/include/cudnn.h 復(fù)制到 /usr/local/cuda/include

然后更新libcudnn.so.7 連接, 然后ldconfig --> <meta charset="utf-8">搜索出可共享的動態(tài)鏈接庫(格式如lib.so),進(jìn)而創(chuàng)建出動態(tài)裝入程序(ld.so)所需的連接和緩存文件。

sudo ln -sf libcudnn.so.7.0.5 libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig -v

到NVIDIA給的sample 里面測試一下

cd samples && make

等待一會之后,編譯完成。

./bandwidthtest

image
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

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