HugeCTR
HugeCTR copied to clipboard
[Requirement] Install version specified dependencies for CUDF in dev.Dockerfile
Currently we don't specify the version for dependencies of CUDF in dev.Dockerfile, which may be at risk when one of them updated by Ubuntu.
Code to install dependencies for cuDF
:
apt update && apt install -y libarrow-dev libarrow-cuda-dev && \
dpkg -r --force-depends libnvidia-compute-450-server
Once cuDF-0.16
is able to be installed by conda
, we should try conda
instead.
We are now using cudf installed from upstream merlin-base image.