使用docker-compose中的nvidia-docker

Question

我想用docker-compose運行2個docker鏡像。

一個圖像應該與nvidia-docker一起運行，另一個應該與docker一起運行。

我已經看到了這個帖子use nvidia-docker-compose launch a container, but exited soon 但是這不是爲我工作（甚至沒有隻運行一個圖片）...

任何想法將是巨大的。

Answer 1

UPDATE：請檢查NVIDIA-泊塢窗2和其支持的碼頭工人，組成第一 https://github.com/NVIDIA/nvidia-docker/wiki/Frequently-Asked-Questions#do-you-support-docker-compose

（我第一次建議增加的NVIDIA-泊塢窗標記）。

如果您查看nvidia-docker-compose代碼here，它只會在localhost：3476上的nvidia配置查詢後爲docker-compose生成特定的docker-file。

你也可以手工製作這個docker-compose文件，因爲它們變得非常簡單，按照這個例子，用你的nvidia驅動程序版本替換375.66，並放置儘可能多的/ dev/nvidia [n]行圖形卡（沒有嘗試把服務在不同的GPU，但去了！）：

services: 
    exampleservice0: 
    devices: 
    - /dev/nvidia0 
    - /dev/nvidia1 
    - /dev/nvidiactl 
    - /dev/nvidia-uvm 
    - /dev/nvidia-uvm-tools 
    environment: 
    - EXAMPLE_ENV_VARIABLE=example 
    image: company/image 
    volumes: 
    - ./disk:/disk 
    - nvidia_driver_375.66:/usr/local/nvidia:ro 
version: '2' 
volumes: 
    media: null 
    nvidia_driver_375.66: 
    external: true 

然後只需運行採用了經典的碼頭工人，排字命令這個手工製作的碼頭工人，撰寫文件。

也許你可以通過跳過其他服務中的nvidia特定的東西來構建非nvidia碼頭工。