安裝完caffe,繼續安裝Digits 。
DIGITS 是英偉達爲普及深度學習開發的圖形化操作界面,簡單易用,旨在幫助初學者跨越入門障礙,迅速上手。因此,DLI 的入門培訓均建議學員從 DIGITS 起步。 說白了, DIGITS 就是一個新手工具。
一、安裝依賴項
sudo apt install --no-install-recommends git graphviz python-dev python-flask python-flaskext.wtf python-gevent python-h5py python-numpy python-pil python-scipy python-tk
二、下載安裝DIGITS
git clone https://github.com/NVIDIA/Digits
sudo -H pip install -r Digits/requirements.txt
sudo -H pip install -e Digits/
三、配置環境變量
sudo nano ~/.bashrc
export CAFFE_ROOT=剛剛安裝caffe的路徑
source ~/.bashrc
四、運行digits-devserver
出現如下錯誤:
解決方法:
sudo -H pip install flask-wtf==0.14.3 -i https://pypi.doubanio.com/simple
flask-wtf 需要安裝新版本