Skip to content

Commit ba4b841

Browse files
committed
update the build from source document and dependency list
1 parent 1179efd commit ba4b841

File tree

3 files changed

+34
-8
lines changed

3 files changed

+34
-8
lines changed

README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ drun rocm/tensorflow:latest
5757
We maintain `tensorflow-rocm` whl packages on PyPI [here](https://pypi.org/project/tensorflow-rocm), to install tensorflow-rocm package using pip:
5858
```
5959
# Install some ROCm dependencies
60-
sudo apt install rocm-libs hipcub miopen-hip
60+
sudo apt install rocm-libs rccl
6161
6262
# Pip3 install the whl package from PyPI
6363
pip3 install --user tensorflow-rocm --upgrade

rocm_docs/tensorflow-build-from-source.md

+31-4
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,12 @@ export ROCM_TOOLKIT_PATH=$ROCM_PATH
2222

2323
## Install required python packages
2424

25-
On Python 3-based systems:
25+
Install the following python dependencies:
2626
```
2727
sudo apt-get update && sudo apt-get install -y \
28+
python-dev \
29+
python-pip \
30+
python-wheel \
2831
python3-numpy \
2932
python3-dev \
3033
python3-wheel \
@@ -34,9 +37,13 @@ sudo apt-get update && sudo apt-get install -y \
3437
python3-yaml \
3538
python3-setuptools && \
3639
sudo apt-get clean
40+
41+
pip3 install keras_preprocessing setuptools keras_applications jupyter --upgrade
42+
pip3 install numpy==1.18.5
3743
```
3844

39-
## Install bazel
45+
## Build Tensorflow 1.15 release
46+
### Install bazel
4047

4148
```
4249
echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list
@@ -46,12 +53,32 @@ cd ~/ && wget https://github.com/bazelbuild/bazel/releases/download/0.24.1/bazel
4653
sudo bash ~/bazel*.sh
4754
```
4855

49-
## Build TensorFlow ROCm port
56+
### Build TensorFlow 1.15 ROCm backend
5057

5158
```
52-
# Clone it
59+
# Clone tensorflow source code
5360
cd ~ && git clone -b r1.15-rocm-enhanced https://github.com/ROCmSoftwarePlatform/tensorflow-upstream.git tensorflow
5461
5562
# Python 3: Build and install TensorFlow ROCm port pip3 package
5663
cd ~/tensorflow && ./build_rocm_python3
5764
```
65+
66+
## Build Tensorflow 2.3 release
67+
### Install bazel
68+
69+
```
70+
echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list
71+
curl https://bazel.build/bazel-release.pub.gpg | sudo apt-key add -
72+
sudo apt-get update && sudo apt-get install -y openjdk-8-jdk openjdk-8-jre unzip wget git
73+
cd ~ && rm -rf bazel*.sh && wget https://github.com/bazelbuild/bazel/releases/download/3.1.0/bazel-3.1.0-installer-linux-x86_64.sh && bash bazel*.sh && rm -rf ~/*.sh
74+
```
75+
76+
### Build TensorFlow 2.3 ROCm backend
77+
78+
```
79+
# Clone tensorflow source code
80+
cd ~ && git clone -b r2.3-rocm-enhanced https://github.com/ROCmSoftwarePlatform/tensorflow-upstream.git tensorflow
81+
82+
# Build and install TensorFlow ROCm port pip3 package
83+
cd ~/tensorflow && ./build_rocm_python3
84+
```

rocm_docs/tensorflow-install-basic.md

+2-3
Original file line numberDiff line numberDiff line change
@@ -61,12 +61,11 @@ Install ROCm pkgs:
6161
```
6262
sudo apt-get update && \
6363
sudo apt-get install -y --allow-unauthenticated \
64-
rocm-dkms rocm-dev rocm-libs hipcub rccl \
64+
rocm-dkms rocm-dev rocm-libs rccl \
6565
rocm-device-libs \
6666
hsa-ext-rocr-dev hsakmt-roct-dev hsa-rocr-dev \
6767
rocm-opencl rocm-opencl-dev \
68-
rocm-utils \
69-
miopen-hip miopengemm
68+
rocm-utils
7069
```
7170

7271
Add username to 'video' group and reboot:

0 commit comments

Comments
 (0)