@@ -22,9 +22,12 @@ export ROCM_TOOLKIT_PATH=$ROCM_PATH
22
22
23
23
## Install required python packages
24
24
25
- On Python 3-based systems :
25
+ Install the following python dependencies :
26
26
```
27
27
sudo apt-get update && sudo apt-get install -y \
28
+ python-dev \
29
+ python-pip \
30
+ python-wheel \
28
31
python3-numpy \
29
32
python3-dev \
30
33
python3-wheel \
@@ -34,9 +37,13 @@ sudo apt-get update && sudo apt-get install -y \
34
37
python3-yaml \
35
38
python3-setuptools && \
36
39
sudo apt-get clean
40
+
41
+ pip3 install keras_preprocessing setuptools keras_applications jupyter --upgrade
42
+ pip3 install numpy==1.18.5
37
43
```
38
44
39
- ## Install bazel
45
+ ## Build Tensorflow 1.15 release
46
+ ### Install bazel
40
47
41
48
```
42
49
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
46
53
sudo bash ~/bazel*.sh
47
54
```
48
55
49
- ## Build TensorFlow ROCm port
56
+ ### Build TensorFlow 1.15 ROCm backend
50
57
51
58
```
52
- # Clone it
59
+ # Clone tensorflow source code
53
60
cd ~ && git clone -b r1.15-rocm-enhanced https://github.com/ROCmSoftwarePlatform/tensorflow-upstream.git tensorflow
54
61
55
62
# Python 3: Build and install TensorFlow ROCm port pip3 package
56
63
cd ~/tensorflow && ./build_rocm_python3
57
64
```
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
+ ```
0 commit comments