8000 Merge pull request #765 from sunway513/develop-upstream-doc-rocm3.0 · darkbuck/tensorflow@024c374 · GitHub
[go: up one dir, main page]
Skip to content

Commit 024c374

Browse files
sunway513sunway513
authored
Merge pull request tensorflow#765 from sunway513/develop-upstream-doc-rocm3.0
Develop upstream doc rocm3.0
2 parents 64c99a7 + b2de51e commit 024c374

File tree

4 files changed

+11
-10
lines changed

4 files changed

+11
-10
lines changed

‎README.ROCm.md

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,6 @@ This repository hosts the port of [Tensorflow](https://github.com/tensorflow/ten
88

99
For further background information on ROCm, refer [here](https://github.com/RadeonOpenCompute/ROCm/blob/master/README.md).
1010

11-
The project is derived from TensorFlow 1.14.0 and has been verified to work with the latest ROCm2.6 release.
12-
1311
For details on installation and usage, see these links:
1412
* [Basic installation](rocm_docs/tensorflow-install-basic.md)
1513
* [Building from source](rocm_docs/tensorflow-build-from-source.md)

‎README.md

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,12 +30,11 @@ to
3030
See all the [mailing lists](https://www.tensorflow.org/community/forums).
3131

3232
**Tensorflow ROCm port**
33-
This project is based on TensorFlow 1.14.0. It has been verified to work with the latest ROCm2.6 release.
3433
Please follow the instructions [here](https://github.com/RadeonOpenCompute/ROCm-docker/blob/master/quick-start.md) to set up your ROCm stack.
3534
A docker container: **rocm/tensorflow:latest(https://hub.docker.com/r/rocm/tensorflow/)** is readily available to be used:
3635
```
37-
alias drun='sudo docker run -it --network=host --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $HOME/dockerx:/dockerx'
38-
drun rocm/tensorflow
36+
alias drun='sudo docker run -it --network=host --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size 16G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $HOME/dockerx:/dockerx'
37+
drun rocm/tensorflow:latest
3938
```
4039
We maintain `tensorflow-rocm` whl packages on PyPI [here](https://pypi.org/project/tensorflow-rocm), to install tensorflow-rocm package using pip:
4140
```

‎rocm_docs/tensorflow-install-basic.md

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -18,10 +18,10 @@ Add the ROCm repository:
1818
wget -qO - http://repo.radeon.com/rocm/apt/debian/rocm.gpg.key | sudo apt-key add -
1919
sudo sh -c 'echo deb [arch=amd64] http://repo.radeon.com/rocm/apt/debian/ xenial main > /etc/apt/sources.list.d/rocm.list'
2020
```
21-
Tensorflow CSB nigtly build requires to ROCm2.6, use the follwoing ROCm repository instead:
21+
Tensorflow CSB nigtly build requires ROCm2.8, use the follwoing ROCm repository instead:
2222
```
2323
wget -qO - http://repo.radeon.com/rocm/apt/debian/rocm.gpg.key | sudo apt-key add -
24-
sudo sh -c 'echo deb [arch=amd64] http://repo.radeon.com/rocm/apt/2.6/ xenial main > /etc/apt/sources.list.d/rocm.list'
24+
sudo sh -c 'echo deb [arch=amd64] http://repo.radeon.com/rocm/apt/2.8.0/ xenial main > /etc/apt/sources.list.d/rocm.list'
2525
```
2626

2727
Install misc pkgs:
@@ -114,10 +114,10 @@ Link to the upstream Tensorflow CSB doc:
114114
We provide nightly tensorflow-rocm whl packages for Python 2.7, 3.5, 3.6 and 3.7 based systems.
115115
After downloading the compatible whl package, you can use pip/pip3 to install.
116116

117-
For example, the following commands can be used to download and install the tensorflow-rocm CSB package on an Ubuntu 16.04 system previously configured with ROCm and Python3.5:
117+
For example, the following commands can be used to download and install the tensorflow-rocm nightly CSB package on an Ubuntu 16.04 system previously configured with ROCm2.8 and Python3.5:
118118
```
119-
wget http://ml-ci.amd.com:21096/job/tensorflow-rocm-release/lastSuccessfulBuild/artifact/pip35_test/whl/tensorflow_rocm-1.14.0-cp35-cp35m-manylinux1_x86_64.whl
120-
pip3 install --user tensorflow_rocm-1.14.0-cp35-cp35m-manylinux1_x86_64.whl
119+
wget http://ml-ci.amd.com:21096/job/tensorflow-rocm-nightly/lastSuccessfulBuild/artifact/pip35_test/whl/tensorflow_rocm-2.0.0-cp35-cp35m-manylinux1_x86_64.whl
120+
pip3 install --user tensorflow_rocm-2.0.0-cp35-cp35m-manylinux1_x86_64.whl
121121
```
122122

123123
## Install TensorFlow ROCm release build

‎rocm_docs/tensorflow-rocm-release.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,15 +7,19 @@ This document is to list the ROCm compatibility for tensorflow-rocm release pack
77

88
| tensorflow-rocm | ROCm base | Installation Instruction |
99
| ------ | ------ | -------- |
10+
| 2.1.0RC1 | ROCm3.0 | <https://pypi.org/project/tensorflow-rocm/2.1.0rc1> |
1011
| 2.1.0RC0 | ROCm2.10 | <https://pypi.org/project/tensorflow-rocm/2.1.0rc0> |
12+
| 2.0.2 | ROCm3.0 | <https://pypi.org/project/tensorflow-rocm/2.0.2> |
1113
| 2.0.1 | ROCm2.10 | <https://pypi.org/project/tensorflow-rocm/2.0.1> |
1214
| 2.0.0 | ROCm2.9 | <https://pypi.org/project/tensorflow-rocm/2.0.0> |
1315
| 2.0.0b3 | ROCm2.7 | <https://pypi.org/project/tensorflow-rocm/2.0.0b3> |
1416
| 2.0.0b2 | ROCm2.6 | <https://pypi.org/project/tensorflow-rocm/2.0.0b2> |
1517
| 2.0.0b1 | ROCm2.6 | <https://pypi.org/project/tensorflow-rocm/2.0.0b1> |
18+
| 1.15.1 | ROCm3.0 | <https://pypi.org/project/tensorflow-rocm/1.15.1> |
1619
| 1.15.0 | ROCm2.10 | <https://pypi.org/project/tensorflow-rocm/1.15.0> |
1720
| 1.15.0rc2 | ROCm2.9 | <https://pypi.org/project/tensorflow-rocm/1.15.0rc2> |
1821
| 1.15.0rc1 | ROCm2.8 | <https://pypi.org/project/tensorflow-rocm/1.15.0rc1> |
22+
| 1.14.5 | ROCm3.0 | <https://pypi.org/project/tensorflow-rocm/1.14.5> |
1923
| 1.14.4 | ROCm2.10 | <https://pypi.org/project/tensorflow-rocm/1.14.4> |
2024
| 1.14.3 | ROCm2.9 | <https://pypi.org/project/tensorflow-rocm/1.14.3> |
2125
| 1.14.2 | ROCm2.8 | <https://pypi.org/project/tensorflow-rocm/1.14.2> |

0 commit comments

Comments
 (0)
0