You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi,我在anachonda3中完成环境配置,尝试运行detect.py时遇到了以下问题,这些问题可能来源于torch版本或者其他的冲突
CUDA=10.0, cudnn=7.6.5, python=3.7, opencv-python=4.2.0.34
测试环境1:torch=1.6, torchvision=0.7, 其他包均从requirement.txt文件安装,运行detect.py报错:
TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
测试环境2:torch=1.7, torchvision=0.8.1, 其他包均遵从U版yolov5 requirement.txt,降低opencv-python版本至4.2。i.e.标准yolov5环境,并测试通过yolov5。加装shapely。运行detect.py报错:
torch.nn.modules.module.ModuleAttributeError: 'Hardswish' object has no attribute 'inplace'
根据网上说法,这个问题需将torch降低至1.6,所以该问题死循环了。
The text was updated successfully, but these errors were encountered:
Hi,我在anachonda3中完成环境配置,尝试运行detect.py时遇到了以下问题,这些问题可能来源于torch版本或者其他的冲突
CUDA=10.0, cudnn=7.6.5, python=3.7, opencv-python=4.2.0.34
测试环境1:torch=1.6, torchvision=0.7, 其他包均从requirement.txt文件安装,运行detect.py报错:
TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
测试环境2:torch=1.7, torchvision=0.8.1, 其他包均遵从U版yolov5 requirement.txt,降低opencv-python版本至4.2。i.e.标准yolov5环境,并测试通过yolov5。加装shapely。运行detect.py报错:
torch.nn.modules.module.ModuleAttributeError: 'Hardswish' object has no attribute 'inplace'
根据网上说法,这个问题需将torch降低至1.6,所以该问题死循环了。
The text was updated successfully, but these errors were encountered: