How to setup a server (including ubuntu / Nvidia GPU)

1. install ubuntu 20.04

        

2. install ssh and nettool

        sudo apt install openssh-client ssh net-tools

3. install Nvidia driver 535:

        sudo apt update -y && sudo apt upgrade -y && sudo apt install vim gcc g++ make python3-pip -y

        use nvidia-detector or ubuntu-drivers devices to check available version

        sudo apt install nvidia-driver-535 -y

        use nvidia-smi to check if it install successfully

4. install docker-nvidia:

curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \ && curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \ sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \ sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list \ && \ sudo apt-get update

sudo apt-get install -y nvidia-container-toolkit

sudo nvidia-ctk runtime configure --runtime=docker

sudo systemctl restart docker

docker run --rm --runtime=nvidia --gpus all ubuntu nvidia-smi