本頁說明如何建立及設定本機深度學習容器。本指南假設您已熟悉 Docker 基本概念。
事前準備
完成下列步驟以設定 Google Cloud 帳戶、啟用必要的 API,以及安裝並啟動必要的軟體。
在 Google Cloud 控制台中,前往「管理資源」頁面,然後選取或建立專案。
-
如果您使用的是 Ubuntu、Debian 等以 Linux 為基礎的作業系統,請將您的使用者名稱加入
docker
群組,這樣就能在不使用sudo
的情況下執行 Docker:sudo usermod -a -G docker ${USER}
將自己加入
docker
群組後,您可能需要重新啟動系統。 開啟 Docker。如要確認 Docker 正在運作,請執行下列 Docker 指令,這個指令會傳回目前的時間和日期:
docker run busybox date
使用
gcloud
做為 Docker 的憑證輔助程式:gcloud auth configure-docker
選用:如要在本機環境中使用 GPU 來執行容器,請安裝
nvidia-docker
。
建立容器
請按照下列步驟建立容器。
如要查看可用的容器清單:
gcloud container images list \ --repository="gcr.io/deeplearning-platform-release"
建議您前往「選擇容器」頁面,選取所需容器。
如果不需要使用支援 GPU 的容器,請輸入下列程式碼範例。將 tf-cpu.1-13 替換為要使用的容器名稱。
docker run -d -p 8080:8080 -v /path/to/local/dir:/home/jupyter \ gcr.io/deeplearning-platform-release/tf-cpu.1-13
如要使用支援 GPU 的容器,請輸入下列程式碼範例。將 tf-gpu.1-13 替換為要使用的容器名稱。
docker run --runtime=nvidia -d -p 8080:8080 -v /path/to/local/dir:/home/jupyter \ gcr.io/deeplearning-platform-release/tf-gpu.1-13
這個指令會在卸離模式中啟動容器,將本機目錄 /path/to/local/dir
掛接到容器中的 /home/jupyter
,並將容器的通訊埠 8080 對應至本機電腦的通訊埠 8080。容器已預先設定為啟動 JupyterLab 伺服器,您可以在 http://localhost:8080
存取該伺服器。
後續步驟
- 如要進一步瞭解如何使用容器,請參閱 Docker 說明文件。