docker pull nvidia/cuda:11.1-cudnn8-devel-ubuntu20.04
docker run -it --name build-onnxruntime-gpu --runtime nvidia \
-v $(pwd)/onnxruntime:/onnxruntime -w /onnxruntime \
nvidia/cuda:11.1-cudnn8-devel-ubuntu20.04
sed -i 's/archive.ubuntu.com/mirrors.aliyun.com/g' /etc/apt/sources.list
apt-get update
apt-get install language-pack-en git cmake python3 python3-pip -y
locale-gen en_US.UTF-8
update-locale LANG=en_US.UTF-8
pip3 config set global.index-url https://mirrors.aliyun.com/pypi/simple/
pip3 install numpy
./build.sh --parallel --build_shared_lib --enable_pybind --build_wheel \
--use_cuda --cudnn_home /usr/include/x86_64-linux-gnu/ \
--cuda_home /usr/local/cuda/ --config Release