message: Switch Dockerfile.rocm to use ROCm 4.0.1
diff: FROM ubuntu:bionic
  MAINTAINER Jeff Poznanovic <jeffrey.poznanovic@amd.com>
  -ARG ROCM_DEB_REPO=http://repo.radeon.com/rocm/apt/4.0/
  +ARG ROCM_DEB_REPO=http://repo.radeon.com/rocm/apt/4.0.1/
  ARG ROCM_BUILD_NAME=xenial
  ARG ROCM_BUILD_NUM=main
  -ARG ROCM_PATH=/opt/rocm-4.0.0
  +ARG ROCM_PATH=/opt/rocm-4.0.1
  ENV DEBIAN_FRONTEND noninteractive
  ENV TF_NEED_ROCM 1
  