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

代码及环境配置均来自【扫盲】VitDet目标检测网络训练

ViTDet程序调用不了GPU全部内存,而且显示NVCC: not available_第1张图片

环境均与视频中相同,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

ViTDet程序调用不了GPU全部内存,而且显示NVCC: not available_第2张图片

 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不可用

你可能感兴趣的:(深度学习,目标检测,人工智能,transformer)