如何在Container Station中使用 docker-compose 來建立可使用 GPU 的 container 呢?

如何在Container Station中使用 docker-compose 來建立可使用 GPU 的 container 呢?

請參考以下指令:

docker run -it --device /dev/nvidia0:/dev/nvidia0 --device /dev/nvidia1:/dev/nvidia1 --device /dev/nvidiactl:/dev/nvidiactl --device /dev/nvidia-uvm:/dev/nvidia-uvm -v /share/CACHEDEV1_DATA/.qpkg/NVIDIA_GPU_DRV/usr/:/usr/local/nvidia tensorflow/tensorflow:1.11.0-gpu-py3 bash

說明：

(1) 以上是兩片顯卡，若只有一片，則移除 --device /dev/nvidia1:/dev/nvidia1。若有超過兩片則自行補上即可，例如 --device /dev/nvidia2:/dev/nvidia2。

(2) 其餘的 --device /dev/nvidiactl:/dev/nvidiactl --device /dev/nvidia-uvm:/dev/nvidia-uvm -v /share/CACHEDEV1_DATA/.qpkg/NVIDIA_GPU_DRV/usr/:/usr/local/nvidia 則固定即可。

若成功設定，進入 container 後使用 nvidia-smi 指令應可看到顯卡資訊。

相關資訊：

• QNAP NAS 搭配 QTS 4.3.5+ 怎麼樣設定GPU
• 如何在QNAP NAS上使用TensorFlow