安裝驅動

驅動使用Lambda Stack

install

LAMBDA_REPO=$(mktemp) && \
wget -O${LAMBDA_REPO} https://lambdalabs.com/static/misc/lambda-stack-repo.deb && \
sudo dpkg -i ${LAMBDA_REPO} && rm -f ${LAMBDA_REPO} && \
sudo apt-get update && sudo apt-get install -y lambda-stack-cuda
sudo reboot

update PyTorch、 TensorFlow、 CUvDA or cuDNN

sudo apt-get update && sudo apt-get dist-upgrade

驗證是否安裝成功

nvidia-smi

安裝 Docker2

sudo apt install -y apt-transport-https ca-certificates curl gnupg-agent software-properties-common &&\
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add - && \
sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable" && \
sudo apt update && \
sudo apt-get install docker-ce docker-ce-cli containerd.io && \
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) && \
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - && \
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list && \
sudo apt update && \
sudo apt install -y nvidia-docker2 && \
sudo pkill -SIGHUP dockerd && \
sudo systemctl stop docker && \
sudo systemctl start docker

docker 加入使用者

sudo usermod -aG docker ${USER} && \
su - ${USER}

sudo usermod -aG docker ${USER} && \

su – ${USER}

測試 gpu

sudo docker run --gpus all nvidia/cuda:11.1-base nvidia-smi

測試啟動年齡性別

docker run --gpus all -d -p 5000:5000 --name mydocker --restart=always raidavid/rai_ai_age_gender:latest

測試啟動 yolo

docker run --gpus all -d -it -p 6000:5000 --name server --restart=always raiaiserver:latest