qq_36848732的博客

TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

1. 安装yolo v5 1.1 clone代码到自己的文件夹下 cd $INSTALL_DIR git clone https://github.com/ultralytics/yolov5.git https://github.com/ultralytics/yolov5 1.2 根据requirements.txt安装依赖 博主自己的环境是ubuntu 16.04，Driver Version: 430.50 ，CUDA Version: 10.1