Skip to content

Commit 32b915e

Browse files
committed
Keep RDMAV_FORK_SAFE=1
1 parent 9ab7273 commit 32b915e

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

pytorch/training/docker/1.8/py3/cu111/Dockerfile.gpu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ ENV DGLBACKEND=pytorch
4040
ENV CMAKE_PREFIX_PATH="$(dirname $(which conda))/../"
4141
ENV SAGEMAKER_TRAINING_MODULE=sagemaker_pytorch_container.training:main
4242
ENV MANUAL_BUILD=0
43+
ENV RDMAV_FORK_SAFE=1
4344

4445
ARG PT_TRAINING_URL=https://aws-pytorch-binaries.s3-us-west-2.amazonaws.com/r1.8.1_aws/20210325-012734/e1343088f0beb99438343e1e99e8d71ffb972b47/gpu/torch-1.8.1-cp36-cp36m-manylinux1_x86_64.whl
4546
ARG PT_TORCHVISION_URL=https://torchvision-build.s3-us-west-2.amazonaws.com/1.8.1/gpu/torchvision-0.9.1-cp36-cp36m-linux_x86_64.whl

tensorflow/training/docker/2.4/py3/cu110/Dockerfile.gpu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ ENV KMP_AFFINITY=granularity=fine,compact,1,0
2121
ENV KMP_BLOCKTIME=1
2222
ENV KMP_SETTINGS=0
2323
ENV MANUAL_BUILD=0
24+
ENV RDMAV_FORK_SAFE=1
2425

2526
ARG PYTHON=python3.7
2627
ARG PYTHON_PIP=python3-pip

0 commit comments

Comments
 (0)