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

问题描述

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

• main.yml：包含我项目的依赖项以及pytorch仅CPU版本
• 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”标准的正确方法，有人知道它是否可行吗？

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

在pytorch channel site上有一个pytorch-cpu软件包，但是它的版本已经过时了（1.1.0，而当前的main是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

解决方法

万一其他人正在寻找答案-我尝试了AMC在上面的评论中建议的内容。我可以确认添加以下行：

- cpuonly

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