Commit 3fdb8db9 authored by lcskrishna's avatar lcskrishna
Browse files

update readme and minor changes

parent 8efd60b2
...@@ -119,7 +119,7 @@ See the [Docker example folder](https://github.com/NVIDIA/apex/tree/master/examp ...@@ -119,7 +119,7 @@ See the [Docker example folder](https://github.com/NVIDIA/apex/tree/master/examp
* Python 3.6 * Python 3.6
* Pytorch 1.5 or newer, The HIPExtensions require 1.5 or newer. * 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. * We recommend follow the instructions from [ROCm-Pytorch](https://github.com/ROCmSoftwarePlatform/pytorch) to install pytorch on ROCm.
* Note: For Pytorch versions less than 1.8 please consider release package [ROCm-Apex v0.3](https://github.com/ROCmSoftwarePlatform/apex/releases/tag/v0.3) . For pytorch versions >= 1.8, use the master branch as this contains latest hipify revamp changes. * Note: For pytorch versions < 1.8, building from source is no longer supported, please use the release package [ROCm-Apex v0.3](https://github.com/ROCmSoftwarePlatform/apex/releases/tag/v0.3) .
# Quick Start # Quick Start
......
...@@ -6,8 +6,6 @@ import sys ...@@ -6,8 +6,6 @@ import sys
import warnings import warnings
import os import os
from torch.utils.hipify import hipify_python
# ninja build does not work unless include_dirs are abs path # ninja build does not work unless include_dirs are abs path
this_dir = os.path.dirname(os.path.abspath(__file__)) this_dir = os.path.dirname(os.path.abspath(__file__))
......
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