У меня есть файл конфигурации среды Conda, который выглядит следующим образом:
name: something
dependencies:
- tensorflow-gpu
- cudatoolkit
- foo
- bar
- ...
- baz
Теперь, в некоторых системах (в тех, которые не имеют графического процессора с поддержкой CUDA), среда выглядела бы так:
name: something
dependencies:
- tensorflow
- foo
- bar
- ...
- baz
Я не уверен, смогу ли я справиться с этим правильно с Кондой. С pip
и requirements-gpu.txt
я могу легко включить другой файл requirements-common.txt
, но я не нашел этот вариант для Conda.
- Можно ли где-нибудь указать "если", чтобы проверить наличие GPU + CUDA и решить, какие зависимости установить?
- Могу ли я поместить "если" где-нибудь, что зависит от аргумента, переданного команде
conda env create
? (т.е.: --flag cpu
или --flag gpu
)
- Можно ли извлечь общие зависимости и каким-то образом включить их из двух разных файлов конфигурации среды верхнего уровня? (т. е.
environment-cpu.yaml
, environment-gpu.yaml
и environment-common.yaml
)
- Может быть что-то другое?