执行:
python train.py --data data/face.data --cfg cfg/yolov3.cfg --epochs 10 --device 0
报错:AssertionError: CUDA unavailable, invalid device 0 requested
原因:安装的pytorch版本不支持GPU
解决方法:
(1)卸载pytorch
pip uninstall torch
pip uninstall torchvision
(2)安装(参考之前写的pytorch安装)
conda install pytorch torchvision cudatoolkit=9.0
(3)查看TensorFlow能否使用GPU
tensorflow.test.is_gpu_available()
输出为False,将TensorFlow版本由1.14.0降为1.10.0就解决了