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、安装CenterNet_Lite 1.1 环境配置 先给出源码git地址：https://github.com/yjh0410/CenterNet-Lite 我的服务器的环境：ubuntu16.04 、CUDA Version 9.0.176、CUDNN 7.5.0 我使用了py3.6，pytorch==1.2.0 与之匹配的torchvision==0.4.0 首先创建了一个新的虚拟环境： conda create -n CenterNetLite python=3.6 source.