Fixed issue of libcuda.so.1 not found
export INSTALL_PREFIX=${CONDA_PREFIX}
+RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/libcuda.so.1
+RUN LD_LIBRARY_PATH=/usr/local/cuda/lib64/stubs/:$LD_LIBRARY_PATH
+
cd ${SRC_DIR}/rmm/python
$PYTHON setup.py build_ext --inplace
$PYTHON setup.py install
+
+RUN rm /usr/local/cuda/lib64/stubs/libcuda.so.1
