Contents
1. Install docker
https://docs.docker.com/engine/install/ubuntu/
2. Setup Nvidia GPU
https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html
3. Clone docker image
https://hub.docker.com/r/pytorch/pytorch/tags
4.Run with GPU
docker run -t -i --gpus all pytorch/pytorch -v /home/iot01/workspace:/workspace
docker run --gpus all -it --shm-size=16gb --net=host --privileged --env="NVIDIA_DRIVER_CAPABILITIES=all" --env="DISPLAY" --env="QT_X11_NO_MITSHM=1" --name="xxx" -v /home/iot01/workspace:/workspace NAMEOFIMAGE
xhost +local: