Hi, I got my solutions on Ubuntu 18.
Modify the start.sh
like this
#if [[ $IS_19 == 1 ]]; then
NV_FLAG="--gpus=all"
#else
# NV_FLAG="--runtime=nvidia -e CUDA_VISIBLE_DEVICES=all"
#fi
Because I found my docker’s version is 20.
It will use NV_FLAG="--runtime=nvidia -e CUDA_VISIBLE_DEVICES=all"
, but this will cause the problem.
So I change to NV_FLAG="--gpus=all"