ViTDet程序调用不了GPU全部内存，而且显示NVCC: not available

代码及环境配置均来自【扫盲】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不可用