search

IBM / powerai

This repo contains ancillary information used to assist users of IBM Watson Machine Learning Community Edition. This repo will contain How To's, Readme's, Dockerfiles, etc. that can be consumed by users looking to get started.
BSD 2-Clause "Simplified" License
56 stars 54 forks source link

Detectron2 build for ppc64le with Pytorch 1.14 #248

Open ticlazau opened 4 years ago

ticlazau commented 4 years ago

Hello,

@hartb please include Detectron2 on the powerai channel. Please find bellow instruction to get this build with Pytorch 1.14

conda config --prepend channels https://public.dhe.ibm.com/ibmdl/export/pub/software/server/ibm-ai/conda-early-access/ conda install pytorch=1.4.0=23447.g18a1a27 cudatoolkit-dev gxx_linux-ppc64le=7 gcc_linux-ppc64le=7

git clone https://github.com/facebookresearch/detectron2.git python -m pip wheel -e detectron2

Thank you, FM

hartb commented 4 years ago

Working on a conda recipe for this.

hartb commented 4 years ago

I'm afraid I won't be able to finish this recipe. If others are interested in picking it up, please feel free.

hartb commented 4 years ago

The start of a recipe is available in a branch in my fork:

https://github.com/hartb/powerai/tree/hartb-detectron2

I think it was to the point where it was building locally, but some of the unit tests were failing.

Note one of the patches included there has been accepted upstream, so can eventually be dropped if the recipe is bumped to a newer version: https://github.com/facebookresearch/detectron2/pull/1357