代码及环境配置均来自【扫盲】VitDet目标检测网络训练
环境均与视频中相同，gpu A6000，环境：
#VS2019
#pytorch 1.9.0
#torchvision 0.10.0
#mmcv-full 1.3.9
#mmdet 2.18.0
conda create -n vitdet python=3.8
conda activate vitdet
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
pip install pytest-runner timm
GPU有47G，但是训练的时候调用不了
RuntimeError: CUDA out of memory. Tried to allocate 176.00 MiB (GPU 0; 47.99 GiB total capacity; 46.15 GiB already allocated; 0 bytes free; 46.79 GiB reserved in total by PyTorch)
训练集不到10G，训练10epoch需要6天，求问怎么分配gpu 而且一直显示NVCC不可用
