使用Docker快速搭建Tensorflow開發環境

當我剛開始學習使用scikit-learn時,老是會出現各類各樣的包依賴問題,兜兜轉轉了一遍才所有安裝好,如今的機器學習算法開發者大都使用tensorflow、pytorch來實現本身的想法,但依然會面臨各類包版本和依賴的問題,有一段時間我使用virtulenv來解決python版本的問題。前幾天升級MacOS,發現由於權限問題無法直接安裝tensorflow。這促使我使用同事們常說的Docker來開發。我搭建的步驟以下:python

 

  1. 從Docker官網下載最新的Docker應用程序(Docker Desktop);檢查是否安裝正確
  2. Docker Hub下載Tensorflow Docker;docker pull tensorflow/tensorflow:latest-py3-jupyter
  3. 掛載本地目錄運行Tensorflow Docker中的jupyter notebook;
    docker run -it -u $(id -u):$(id -g) -v /Users/hh/Documents/ghdl:/tf/ghdl -p 8888:8888 --rm tensorflow/tensorflow:latest-py3-jupyter  
  4. 在瀏覽器中輸入notebook網址

相關文章
相關標籤/搜索