本文將示範建立可使用gpu的docker image，使用pytorch深度學習框架。
此image環境為
python:3.7
pytorch:1.9.0
numpy==1.21.5
pandas==1.3.5
matplotlib==3.5.1
scipy==1.7.3
scikit-learn==1.0.2
一、建立一個資料夾，包含兩個檔案
- dockerfile
- requirements.txt
dockerfile內容如下
Remark:
1. dockerfile可以用記事本打開
2. 如果有其他想複製的檔案，可以使用COPY指令複製到/app下，例如: COPY folder /app或 COPY test.py /app
requirements.txt內容如下(可以照自己的需求調整)
二、以命令提示字元進入該資料夾後，執行以下指令
$ docker build -t torch-gpu-1.9:0.1.0 .
Remark: 最後面的點(.)不可以少
三、run container
兩種方法如下:
-v 的意思是要把本地端的某個資料夾link到container內的/app資料夾下，故run起container後，應該能在container內看到本地端的檔案
如下:
四、測試能不能讀到GPU
最後
之後做開發，變可以直接在本地資料夾開發，寫好要跑的script後，run起docker container執行，或是將docker image及script一起給別人~
