個人電腦如何搭建深度學習/機器學習開發環境?


硬件信息:

顯卡GTX 750Ti + 4核處理器 + 16G內存 + 120G固態 + 500G機械

 

軟件信息:

ubuntu16.04:一開始安裝的是18.04,NVIDIA官方暫未提供這個版本的driver,安裝過程中提示找不帶顯卡驅動。建議大家不要盲目追求新。

cuda:GPU上的並行計算平台和模型;版本選擇cuda-8.0

cudnn:相比標准的cuda,它在一些常用的神經網絡操作上進行了性能的優化;版本選擇cudnn 6.0

anaconda:一個開源的Python發行版本,其包含了conda、python等180多個科學包及其依賴項;版本選擇Anaconda3-4.3.0-Linux-x86_64.sh該版本基於Python 3.6

tensorflow:作為keras的backend,本次安裝選擇的是GPU版,由於cuda是8.0,建議tensorflow選擇tensorflow-gpu==1.4.0

keras:基於Python的深度學習庫,是一個高層神經網絡API,后端可以選擇Tensorflow、Theano以及CNTK。

 

安裝步驟:

安裝 ubuntu16.04+win7雙系統:

在win7上壓縮出一個free空間,作為ubuntu系統安裝位置。

使用rufus工具將U盤制作為安裝ubuntu的系統盤,安裝步驟教程有很多。

 

安裝 cuda-8.0

安裝:參開圖中的installation instruction => sudo sh cuda_8.0.44_linux.run

注意:安裝過程中會提示是否安裝顯卡驅動,選擇“否”(n)

檢驗:nvcc --version這種方法不夠准確;建議使用deviceQuery,出現下圖信息表示cuda安裝成功,Result=PASS

添加到環境變量:在~/.bashrc最后面加入下面的代碼

export PATH="/usr/local/cuda-8.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH"

 

安裝 cudnn 6.0

將下載的tgz文件解壓:sudo tar xvf xxx.tgz

分別將解壓后的include、lib64文件夾中的全部文件copy到/usr/local/cuda/include//usr/local/cuda/lib64

重新建立軟連接:

cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.6
sudo chmod u=rwx,g=rx,o=rx libcudnn.so.6.0.21
sudo ln -s libcudnn.so.6.0.21 libcudnn.so.6
sudo ln -s libcudnn.so.6 libcudnn.so

 

安裝 Anaconda3-4.3.0-Linux-x86_64.sh

直接運行下載后的sh腳本:

bash Anaconda3-4.3.0-Linux-x86_64.sh 

根據提示,敲擊enter或輸入yes。

檢驗:一定打開新的終端來使用anaconda

 

安裝tensorflow-gpu

更換pip源:

cd ~
sudo mkdir .pip
cd .pip
sudo touch pip.conf
sudo chmod 777 pip.conf
gedit pip.conf

在打開的pip.conf中加入下面的信息

[global]
index-url = https://pypi.douban.com/simple

更換conda源:

touch ~/.condarc
chmod 777 ~/.condarc
gedit ~/.condarc

在打開的.condarc中加入下面的信息

channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- defaults
show_channel_urls: true

建立conda環境:

conda create -n tensorflow-gpu python=3.6

 激活上一步建立的tensorflow-gpu環境,安裝tensorflow-gpu

source activate tensorflow-gpu
pip install tensorflow-gpu==1.4.0

安裝過程中提示更新pip,按照提示操作即可:

 

安裝keras

在tensorflow-gpu環境下,使用pip安裝:

pip install keras

 

 檢驗tensorflow+keras環境

 

 

參考文章:

https://blog.csdn.net/Super_jm_/article/details/78397074

https://blog.csdn.net/huang826336127/article/details/78754767

https://blog.csdn.net/gangeqian2/article/details/79358543


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM