use entrypoint from 'rapids' environment

rapidsai · Oct 1, 2024 · e149215 · e149215
1 parent 41e77b9
commit e149215
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -103,7 +103,7 @@ conda clean -afy
 EOF
 
 # Disable the JupyterLab announcements
-RUN /opt/conda/bin/jupyter labextension disable "@jupyterlab/apputils-extension:announcements"
+RUN /opt/conda/envs/rapids/bin/jupyter labextension disable "@jupyterlab/apputils-extension:announcements"
 
 ENV DASK_LABEXTENSION__FACTORY__MODULE="dask_cuda"
 ENV DASK_LABEXTENSION__FACTORY__CLASS="LocalCUDACluster"
@@ -140,7 +140,7 @@ LABEL com.nvidia.workbench.package-manager.apt.binary="/usr/bin/apt"
 LABEL com.nvidia.workbench.package-manager.apt.installed-packages=""
 LABEL com.nvidia.workbench.package-manager.conda3.binary="/opt/conda/bin/conda"
 LABEL com.nvidia.workbench.package-manager.conda3.installed-packages="rapids cudf cuml cugraph rmm pylibraft cuspatial cuxfilter cucim xgboost jupyterlab"
-LABEL com.nvidia.workbench.package-manager.pip.binary="/opt/conda/bin/pip"
+LABEL com.nvidia.workbench.package-manager.pip.binary="/opt/conda/envs/rapids/bin/pip"
 LABEL com.nvidia.workbench.package-manager.pip.installed-packages="jupyterlab-nvdashboard"
 LABEL com.nvidia.workbench.programming-languages="python3"
 LABEL com.nvidia.workbench.schema-version="v2"

diff --git a/context/raft-ann-bench/get_datasets.sh b/context/raft-ann-bench/get_datasets.sh
@@ -3,7 +3,7 @@
 
 set -eo pipefail
 
-export CONDA_PREFIX=/opt/conda
+export CONDA_PREFIX=/opt/conda/envs/rapids
 
 python -m raft_ann_bench.get_dataset --dataset deep-image-96-angular --normalize --dataset-path /home/rapids/preloaded_datasets
 python -m raft_ann_bench.get_dataset --dataset fashion-mnist-784-euclidean --dataset-path /home/rapids/preloaded_datasets

diff --git a/context/raft-ann-bench/run_benchmark.sh b/context/raft-ann-bench/run_benchmark.sh
@@ -14,7 +14,7 @@ function hasArg {
     (( ${NUMARGS} != 0 )) && (echo " ${ARGS} " | grep -q " $1 ")
 }
 
-export CONDA_PREFIX=/opt/conda
+export CONDA_PREFIX=/opt/conda/envs/rapids
 export DATASET_ARG=$1
 export GET_DATASET_ARGS=$2
 export RUN_ARGS=$3

diff --git a/context/raft-ann-bench/run_benchmarks_preloaded_datasets.sh b/context/raft-ann-bench/run_benchmarks_preloaded_datasets.sh
@@ -14,7 +14,7 @@ function hasArg {
     (( ${NUMARGS} != 0 )) && (echo " ${ARGS} " | grep -q " $1 ")
 }
 
-export CONDA_PREFIX=/opt/conda
+export CONDA_PREFIX=/opt/conda/envs/rapids
 export DATASET_ARG=$1
 export GET_DATASET_ARGS=$2
 export RUN_ARGS=$3