适用于GPU和非GPU版本的Dockerfile

Question

我有一个使用Dockerfile维护的项目。这是一个数据科学项目，因此docker可以在很多环境中使用，无论有没有GPU，都可以在本地，云中使用。

GPU与非GPU版本之间的差异很小，仅此而已：

FROM ubuntu:18.04  -> non-GPU
FROM nvidia/cuda   -> GPU 

处理这种情况的最佳方法是什么？

PS：我正在使用docker-compose构建docker

Answer 1

您可以在docker-compose yml文件中使用多个docker文件，如下所示，

├── docker-compose.yml
├── GPU project
│   └── GPU.Dockerfile
└── NON-GPU project
    └── NONGPU.Dockerfile

还可以通过提供文件Dockerfile扩展名（而不是名称）来在VS代码中突出显示语法。 GPU.Dockerfile，NONGPU.Dockerfile等。