first commit

2024-09-08 08:58:59 -07:00
commit 54e4478556
8 changed files with 229 additions and 0 deletions
@@ -0,0 +1,20 @@
+ARG OS_VERSION=16.04
+FROM ubuntu:${OS_VERSION}
+ARG ROCM_VERSION=3.5.1
+RUN <<EOF
+apt-get update
+DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends ca-certificates curl gnupg
+curl -sL http://repo.radeon.com/rocm/rocm.gpg.key | apt-key add -
+echo "deb [arch=amd64] http://repo.radeon.com/rocm/apt/${ROCM_VERSION}/ xenial main" > /etc/apt/sources.list.d/rocm.list
+apt-get update
+DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends sudo libelf1 libnuma-dev build-essential git vim-nox cmake-curses-gui kmod file python3 python3-pip rocm-dev
+apt-get clean
+rm -rf /var/lib/apt/lists/*
+EOF
+COPY sudo-nopasswd /etc/sudoers.d/sudo-nopasswd
+COPY path-fix /etc/sudoers.d/path-fix
+RUN useradd --create-home -G sudo,video --shell /bin/bash rocm
+USER rocm
+WORKDIR /home/rocm
+ENV PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/opt/rocm/bin
+CMD ["/bin/bash", "-l"]
@@ -0,0 +1,2 @@
+# This line modifies the secure path
+Defaults secure_path="/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/snap/bin:/opt/rocm/bin"
@@ -0,0 +1,128 @@
+1.9.3	xenial
+2.0	xenial
+2.1	xenial
+2.10.0	xenial
+2.10.0-hipclang	xenial
+2.2	xenial
+2.3	xenial
+2.4	xenial
+2.5	xenial
+2.6	xenial
+2.7	xenial
+2.7.1	xenial
+2.7.2	xenial
+2.8.0	xenial
+2.9.0	xenial
+3.0	xenial
+3.1	xenial
+3.1.1	xenial
+3.10	xenial
+3.3	xenial
+3.5	xenial
+3.5.1	xenial
+3.7	xenial
+3.8	xenial
+3.9	xenial
+3.9.1	xenial
+4.0	xenial
+4.0.1	xenial
+4.1	xenial
+4.1.1	xenial
+4.2	ubuntu
+4.2	xenial
+4.3	ubuntu
+4.3	xenial
+4.3.1	ubuntu
+4.3.1	xenial
+4.5	ubuntu
+4.5.1	ubuntu
+4.5.2	ubuntu
+5.0	ubuntu
+5.0.1	ubuntu
+5.0.2	ubuntu
+5.1	ubuntu
+5.1.1	ubuntu
+5.1.2	ubuntu
+5.1.3	ubuntu
+5.1.4	ubuntu
+5.2	ubuntu
+5.2.1	ubuntu
+5.2.3	ubuntu
+5.2.4	ubuntu
+5.2.5	ubuntu
+5.3	focal
+5.3	jammy
+5.3	ubuntu
+5.3.2	focal
+5.3.2	jammy
+5.3.2	ubuntu
+5.3.3	focal
+5.3.3	jammy
+5.3.3	ubuntu
+5.4	focal
+5.4	jammy
+5.4	ubuntu
+5.4.1	focal
+5.4.1	jammy
+5.4.1	ubuntu
+5.4.2	focal
+5.4.2	jammy
+5.4.2	ubuntu
+5.4.3	focal
+5.4.3	jammy
+5.4.3	ubuntu
+5.4.5	focal
+5.4.5	jammy
+5.4.5	ubuntu
+5.4.6	focal
+5.4.6	jammy
+5.4.6	ubuntu
+5.5	focal
+5.5	jammy
+5.5	ubuntu
+5.5.1	focal
+5.5.1	jammy
+5.5.1	ubuntu
+5.5.2	focal
+5.5.2	jammy
+5.5.2	ubuntu
+5.5.3	focal
+5.5.3	jammy
+5.5.3	ubuntu
+5.6	focal
+5.6	jammy
+5.6	ubuntu
+5.6.1	focal
+5.6.1	jammy
+5.6.1	ubuntu
+5.7	focal
+5.7	jammy
+5.7	ubuntu
+5.7.1	focal
+5.7.1	jammy
+5.7.1	ubuntu
+5.7.2	focal
+5.7.2	jammy
+5.7.2	ubuntu
+5.7.3	focal
+5.7.3	jammy
+5.7.3	ubuntu
+
+6.0	focal
+6.0	jammy
+6.0	ubuntu
+6.0.1	focal
+6.0.1	jammy
+6.0.1	ubuntu
+6.0.2	focal
+6.0.2	jammy
+6.0.2	ubuntu
+6.0.3	focal
+6.0.3	jammy
+6.0.3	ubuntu
+6.1	focal
+6.1	jammy
+6.1	ubuntu
+6.1.1	focal
+6.1.1	jammy
+6.1.1	ubuntu
@@ -0,0 +1,3 @@
+# This line grants the any members of the sudo group
+# passwordless sudo privileges
+%sudo ALL=(ALL) NOPASSWD:ALL
@@ -0,0 +1,17 @@
+FROM ubuntu:22.04
+RUN <<EOF
+apt-get update
+apt-get install -y python3 python3-pip python3-dev python3-wheel build-essential git wget
+wget -O /usr/local/bin/bazel "https://github.com/bazelbuild/bazelisk/releases/download/v1.19.0/bazelisk-linux-amd64" && chmod +x /usr/local/bin/bazel
+git clone https://github.com/tensorflow/tensorflow.git
+EOF
+RUN pip3 install -U --user pip numpy wheel packaging requests
+RUN pip3 install -U --user keras_preprocessing --no-deps
+WORKDIR /tensorflow
+RUN git checkout r2.12
+RUN yes "" | CC_OPT_FLAGS="-march=native" ./configure
+RUN bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package
+RUN ./bazel-bin/tensorflow/tools/pip_package/build_pip_package /tensorflow_pkg
+WORKDIR /tensorflow_pkg
+RUN pip3 install tensorflow*.whl
+CMD ["/bin/bash", "-l"]
@@ -0,0 +1,12 @@
+apt update
+apt upgrade -y
+apt install wget sudo git
+wget -q -O - http://repo.radeon.com/rocm/rocm.gpg.key | apt-key add -
+apt update
+apt install -y rocm-dev rocm-utils rocm-libs miopen-hip miopengemm rccl
+export PATH=/opt/rocm-3.5.1/bin:$PATH
+# python 3.6
+pip install wheel
+python3 setup.py bdist_wheel
+
+echo "gfx803" >> $ROCM_PATH/bin/target.lst'
@@ -0,0 +1,24 @@
+import requests
+from bs4 import BeautifulSoup
+import re
+from multiprocessing import Pool
+import itertools
+
+base = "https://repo.radeon.com/rocm/apt"
+
+r = requests.get(base).text
+versionPage = BeautifulSoup(r, 'html.parser')
+getVersion = lambda e : e.get('href')[:-1]
+versions = list(map(getVersion, versionPage.findAll('a', attrs={'href': lambda v: re.match(r'^\d', v)})))
+
+def checkVersion(v: str):
+    r = requests.get(f"{base}/{v}/dists").text
+    distPage = BeautifulSoup(r, 'html.parser')
+    codes = distPage.findAll('a', attrs={'href': lambda v: not re.match(r'^[http|\.{2}]', v)})
+    return [(v, code.string[:-1]) for code in codes]
+    
+with Pool() as p:
+    results = p.map(checkVersion, versions)
+    for r in results:
+        for (v, c) in r:
+            print(f"{v}\t{c}")