我将conda 4.8.3与Python 3.7配合使用,正在编写环境文件以指定项目的依赖项。我想编写几个文件以能够安装多个环境:
要进行基本(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频道站点上,有一个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
万一其他人正在寻找答案-我尝试了AMC在上面的评论中建议的内容。我可以确认添加以下行:
- cpuonly
我的environment.yml文件强制下载了pytorch的CPU版本。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句