如何在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 指令應可看到顯卡資訊。
相關資訊：
留言
張貼留言
發表一下意見，互動一下唄！