close


如何在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 指令應可看到顯卡資訊。

相關資訊：


喜歡這篇文章 歡迎分享 如果有任何建議 歡迎留言啊 :)
arrow
arrow
    全站熱搜

    tttt 發表在 痞客邦 留言(0) 人氣()

    E-mail轉寄