Skip to content

GitLab

Unverified Commit 8fff447e authored by Chaitanya Sri Krishna Lolla's avatar Chaitanya Sri Krishna Lolla Committed by GitHub
Browse files

Update ReadMe

parent 979551d8
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 5 deletions
README.md
View file @ 8fff447e
...@@ -115,17 +115,18 @@ It's often convenient to use Apex in Docker containers. Compatible options incl ...@@ -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. 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 # Quick Start
### Rocm ### Rocm
Apex on ROCm supports both python only build and extension build. 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. 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 python3.6 setup.py install
``` ```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment