Skip to content

Steps to install Apex on ROCm.

Jump to bottom
Chaitanya Sri Krishna Lolla edited this page Aug 21, 2020 · 2 revisions

This page explains the details on how to install Apex on ROCm.

Pre-Requisites:

  1. Pytorch installed on ROCm: Link
  2. python3.6

Installation Steps:

There are various options to install apex on ROCm. It depends on your use-case to use one of the options.

  1. Install Apex with Python only build

python3.6 setup.py install

  1. Install Apex with CPP Extensions enabled.

python3.6 setup.py install --cpp_ext

  1. Install Apex with HIPExtensions enabled

python3.6 setup.py install --cuda_ext

  1. To use both CPPExtensions and HIPExtensions :

python3.6 setup.py install --cuda_ext --cpp_ext

  1. Install apex via using ninja and directly using pip3.6:

pip3.6 install ninja
pip3.6 install -v --install-option="--cpp_ext" --install-option="--cuda_ext" 'git+https://github.com/ROCmSoftwarePlatform/apex.git'
Clone this wiki locally