Conda环境：几个环境文件-指定cpu-only版本的Pytorch

Question

我正在使用 conda 4.8.3 和 Python 3.7，我正在编写环境文件来指定我的项目的依赖项。我想写几个文件来安装几个环境：

main.yml ：包含我的项目的依赖项，以及 pytorch CPU-only version
dev.yml：包含开发工具（mypy、flake8、pytest ..）
gpu.yml : 包含pytorch-GPU（带有指定版本的CUDA）

要获得基本 (CPU) 安装，可以这样写：conda env update --file main.yml

要获得与 GPU 兼容的安装，则需要添加 conda env update --file gpu.yml

这是我的问题：目前我找不到在环境文件中为 pytorch 指定 'CPU-only' 标准的正确方法，有人知道它是否可行吗？

通常用于此目的的命令是 conda install pytorch torchvision cpuonly -c pytorch，但我找不到在 yml 文件中指定它的方法。

在pytorch channel site上，有一个pytorch-cpu包，但是版本比较旧（1.1.0，目前主要是1.6.0）

这是我的 main.yml 环境文件：

name: my_env
channels:
  - intel
  - conda-forge
  - pytorch

dependencies:

  - numpy
  - scipy
  - scikit-image
  - matplotlib
  - wxpython
  - colorama
  - dill
  - protobuf
  - pytorch   # How to specify the 'cpu' criteria here??
  - torchvision 
  - pip:
      - -r env/requirements.txt

Answer 1

以防其他人正在寻找答案——我尝试了 AMC 在上面的评论中建议的方法。我可以确认添加行：

- cpuonly

我的 environment.yml 文件强制下载 CPU 版本的 pytorch。

Conda环境：几个环境文件-指定cpu-only版本的Pytorch

Conda environment : several environment files - specify cpu-only version of Pytorch

python

conda

pytorch