问题描述
我有一个使用Dockerfile维护的项目。这是一个数据科学项目,因此docker可以在很多环境中使用,无论有没有GPU,都可以在本地,云中使用。
GPU与非GPU版本之间的差异很小,仅此而已:
FROM ubuntu:18.04 -> non-GPU
FROM nvidia/cuda -> GPU
处理这种情况的最佳方法是什么?
PS:我正在使用docker-compose构建docker
解决方法
您可以在docker-compose yml文件中使用多个docker文件,如下所示,
├── docker-compose.yml
├── GPU project
│ └── GPU.Dockerfile
└── NON-GPU project
└── NONGPU.Dockerfile
还可以通过提供文件Dockerfile扩展名(而不是名称)来在VS代码中突出显示语法。 GPU.Dockerfile,NONGPU.Dockerfile等。