技術工具
- 工具首頁\
- MXNet/Dev-GPU
### 容器說明
cu10.0-dnn7.6-gpu-mx-19.09 代表此環境為 2019 年 9 月(19.09) 建置的 GPU 運算開發容器。此環境內含使用 NVIDIA CUDA 10.0 (cu10.0)和 NVIDIA cuDNN 7.6(dnn7.6)的 MXNet(mx) 開發框架。
本環境除包含 CUDA runtime 外,亦有 CUDA development files。因此,本環境亦可以用來編譯/開發 CUDA 程式。
本環境除了單機/單伺服器使用以外,亦可透過 Horovod + OpenMPI 實現叢集伺服器跨節點運算。
### 下載方式
請於終端機執行以下指令:
```bash
docker pull moeidb/aigo:cu10.0-dnn7.6-gpu-mx-19.09
```
### 使用方式
#### 使用範例 1: 於背景啟動 Jupyterlab 服務
掛載當前位置目錄 (```$PWD```) 至容器內部的 ```/workspace``` 資料夾,並且讓 Jupyterlab 服務監聽本機的 port ```9999```:
```bash
# 決定 Jupyterlab 該監聽本機的哪一個 port
host_port=9999
# 啟動容器並取得容器 ID
container_id=$(nvidia-docker run --rm -d -p 9999:8888 -v $PWD:/workspace moeidb/aigo:cu10.0-dnn7.6-gpu-mx-19.09) # 休息一會,靜待容器服務啟動
# 等待服務啟動
sleep 2.
# 擷取容器的 Jupyterlab token
notebook_token=$(docker logs ${container_id} 2>&1 | grep -nP "(LabApp.*token=).*" | cut -d"=" -f 2)
# 顯示連線至 Jupyterlab 服務的網址
printf "Open a browser and connect to:
http://[your_ip]:${host_port}/?token=${notebook_token}
"
```
輸入以上指令於終端機後,應該會顯示一個網址:
```bash
Open a browser and connect to:
http://[your_ip]:9999/?token=87f6f7ad1455b7dde323f8a570897d4bf9dace8659e0e9bd
```
這代表我們已經在容器內啟動了 Jupyterlab 服務。接著,請開啟瀏覽器,並貼上此網址,即可使用 Jupyterlab 來撰寫 Python 筆記本。
注意事項:
* 網址當中可見 ```token=87f6f7ad14...```,其中 ```87f6f7ad14...``` 是亂數產生的一串 token。由於 token 為隨機字串,因此您實際取得的 token 應和本範例不同。
* 需將 ```[your_ip]``` 更改為機器的 IP 位址。 若您於本地端使用,則 ```[your_ip]``` 應為 ```127.0.0.1```。
* 開啟 Jupyterlab 後,會自動進入 ```/workspace``` 資料夾。
* 因為建立容器時已將本機當前目錄 ```${PWD}``` 掛載至容器內的 ```/workspace```,所以您應該會在 ```/workspace``` 內看到先前存放於本機 ```${PWD}``` 的檔案。
#### 使用範例 2: 利用容器環境執行 Python 腳本
```bash
# 建立一個測試腳本。此腳本將單純的匯入 PyMC3 套件,並且印出其當前版本。
printf "import mxnet \
\nprint('MXNet version=', mxnet.__version__)" \
> check_mx_version.py
# 我們已建立一個位於 ${PWD} 的 Python 腳本。接著,我們試跑一個容器來執行它:
nvidia-docker run -it --rm -v ${PWD}:/workspace \
moeidb/aigo:cu10.0-dnn7.6-gpu-mx-19.09 python3 check_mx_version.py
```
輸入以上指令於終端機後,應會顯示出 Docker 容器內部所安裝的 PyTorch 版本,如下:
```bash
MXNet version= 1.6.0
```
請先登入後輸入您的回覆