From 979551d80f1e03589660e314c17e4dd79bb77b2b Mon Sep 17 00:00:00 2001 From: lcskrishna Date: Fri, 29 May 2020 12:18:54 -0700 Subject: [PATCH 1/2] update readme --- README.md | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/README.md b/README.md index e48efee5e..bf92a7647 100644 --- a/README.md +++ b/README.md @@ -117,6 +117,24 @@ See the [Docker example folder](https://github.com/NVIDIA/apex/tree/master/examp # Quick Start +### Rocm +Apex on ROCm supports both python only build and extension build. + +Pre-requisites: +* Pytorch installed on ROCm. + +Note: Pytorch version recommended is >=1.5 for extension build. + +### To install using python only build use the following: +``` +python3.6 setup.py install +``` + +### To install using extensions enabled use the following command in apex folder: +``` +python3.6 setup.py install --cpp_ext --cuda_ext +``` + ### Linux For performance and full functionality, we recommend installing Apex with From 8fff447e42ad004519834334b8f08ce0181da8d6 Mon Sep 17 00:00:00 2001 From: Chaitanya Sri Krishna Lolla Date: Fri, 29 May 2020 13:34:02 -0700 Subject: [PATCH 2/2] Update ReadMe --- README.md | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index bf92a7647..3cd0fa707 100644 --- a/README.md +++ b/README.md @@ -115,17 +115,18 @@ It's often convenient to use Apex in Docker containers. Compatible options incl See the [Docker example folder](https://github.com/NVIDIA/apex/tree/master/examples/docker) for details. +## On ROCm: +* Python 3.6 +* Pytorch 1.5 or newer, The HIPExtensions require 1.5 or newer. +* We recommend follow the instructions from [ROCm-Pytorch](https://github.com/ROCmSoftwarePlatform/pytorch) to install pytorch on ROCm. + # Quick Start ### Rocm Apex on ROCm supports both python only build and extension build. - -Pre-requisites: -* Pytorch installed on ROCm. - Note: Pytorch version recommended is >=1.5 for extension build. -### To install using python only build use the following: +### To install using python only build use the following command in apex folder: ``` python3.6 setup.py install ```