diff --git a/ubuntu18.04/cuda-devel/Dockerfile b/ubuntu18.04/cuda-devel/Dockerfile index 2bc09ea335b5710e5153f333bb8d965775f617d2..0913bc432355907890f592eefbadecd5946dbbb4 100644 --- a/ubuntu18.04/cuda-devel/Dockerfile +++ b/ubuntu18.04/cuda-devel/Dockerfile @@ -28,4 +28,4 @@ RUN apt-get update && \ cd python && \ python3.7 setup.py install && \ apt-get clean -ADD mxnet_profile.sh /etc/profile.d/mxnet.sh +ENV MXNet_DIR /opt/mxnet-1.4.1 diff --git a/ubuntu18.04/devel/Dockerfile b/ubuntu18.04/devel/Dockerfile index 8b664a26911487595e274521a00d99d8f2d7c747..c93b4efc73a6faeae8f7ececfa7758ea5cb1ddf2 100644 --- a/ubuntu18.04/devel/Dockerfile +++ b/ubuntu18.04/devel/Dockerfile @@ -27,4 +27,4 @@ RUN apt-get update && \ cd python && \ python3.7 setup.py install && \ apt-get clean -ADD mxnet_profile.sh /etc/profile.d/mxnet.sh +ENV MXNet_DIR /opt/mxnet-1.4.1 diff --git a/ubuntu18.04/mxnet_profile.sh b/ubuntu18.04/mxnet_profile.sh deleted file mode 100755 index 993fda21d9cc868d2e09df87a4379b7236a5dacf..0000000000000000000000000000000000000000 --- a/ubuntu18.04/mxnet_profile.sh +++ /dev/null @@ -1,2 +0,0 @@ -#! /bin/sh -export MXNet_DIR=/opt/mxnet-1.4.1