solution of “CUDA”，“Ninja”

[weikangqi]

docker is all your need

1. first pull docker image

docker pull pytorch/pytorch:1.10.0-cuda11.3-cudnn8-devel

2. run docker container
   You need to configure it according to your own ???

docker run --gpus all -it  --shm-size="32g" --name=SPIN  -v /home/???/code:/code pytorch/pytorch:1.10.0-cuda11.3-cudnn8-devel/bin/bash

3. in the container
   key steps

cd /code/SPIN
pip install -r requirements.txt
chmod 1777 /tmp
apt  install ffmpeg libsm6 libxext6  -y 
pip install Ninja
python train.py --config ./configs/spin_light_x4.yml