我正在使用Python SDK在Azure机器学习服务中构建环境对象,并且一切正常, 一个从URL安装的Python程序包。我想知道如何处理。这有效:
my_env = Environment.from_conda_specification("trident", './environment.yml')
..但是Docker构建在其中一个从文件安装的软件包中失败。
[91mERROR:找不到满足detectron2 == 0.1.3 + cu101要求的版本(来自-r /azureml-environment-setup/condaenv.s5fi23rw.requirements.txt(第7行))(来自以下版本:没有) [0m [91m错误:找不到用于detectron2 == 0.1.3 + cu101的匹配分布(来自-r /azureml-environment-setup/condaenv.s5fi23rw.requirements.txt(第7行)) [0m [91m
这是我手动安装该软件包的方法:
python -m pip install detectron2 -f / https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.5/index.html
还有另一个应从github安装的软件包,如下所示:
pip install -U 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
我对yaml文件一无所知:是否可以在yaml文件中包含类似的语法?
我希望不必在本地重新构建环境并从中进行安装(这是替代选择),因为这样做必须重新安装CUDA。
谢谢
答案 0 :(得分:-1)
更新是因为谁喜欢否决票,某人可能会觉得这很有用。
相同的规范来安装Conda软件包。OP可能已应用了以下内容:
# run: conda env create --file environment.yml
name: test-env
dependencies:
- python>=3.5
- anaconda
- pip
- pip:
# works for regular pip packages
- docx
- gooey
# for github
- git+https://github.com/facebookresearch/detectron2.git
# and for wheels
- https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/index.html
但是,由于CUDA / CuDNN兼容的乐趣,我发现使用Docker映像将Detectron2加载到AzureML容器上要容易得多。
FROM mcr.microsoft.com/azureml/base-gpu:openmpi3.1.2-cuda10.1-cudnn7-ubuntu18.04
RUN apt update && apt install git -y && rm -rf /var/lib/apt/lists/*
RUN /opt/miniconda/bin/conda update -n base -c defaults conda
RUN /opt/miniconda/bin/conda install -y cython=0.29.15 numpy=1.18.1
# Install cocoapi, required for drawing bounding boxes
RUN git clone https://github.com/cocodataset/cocoapi.git && cd cocoapi/PythonAPI && python setup.py build_ext install
RUN pip install --user tensorboard cython
RUN pip install --user torch==1.5+cu101 torchvision==0.6+cu101 -f https://download.pytorch.org/whl/torch_stable.html
RUN pip install azureml-defaults
RUN pip install azureml-dataprep[fuse]
RUN pip install pandas pyarrow
RUN pip install opencv-python-headless
RUN pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/index.html```