diff --git a/Dockerfile-default-rocm b/Dockerfile-default-rocm index d53ae680..4a57009e 100644 --- a/Dockerfile-default-rocm +++ b/Dockerfile-default-rocm @@ -195,6 +195,10 @@ ENV WITH_RCCL=$WITH_RCCL ARG WITH_NFS_WORKAROUND=1 ENV WITH_NFS_WORKAROUND=$WITH_NFS_WORKAROUND +#MIOPEN_DEBUG_SAVE_TEMP_DIR is required to prevent +# PAD-134 +ENV MIOPEN_DEBUG_SAVE_TEMP_DIR=1 + ENTRYPOINT ["/container/bin/scrape_libs.sh"] CMD ["/bin/bash"] USER root diff --git a/dockerfile_scripts/additional-requirements-rocm.txt b/dockerfile_scripts/additional-requirements-rocm.txt index 27585b8e..f352d49f 100644 --- a/dockerfile_scripts/additional-requirements-rocm.txt +++ b/dockerfile_scripts/additional-requirements-rocm.txt @@ -10,7 +10,7 @@ Pillow>=8.3.2,<=9.5.0 analytics-python nvidia-ml-py protobuf<=3.20.3 -tensorboard==2.10.1 +tensorboard==1.15 pynvml tokenizers==0.13.0 huggingface-hub==0.16.4