在使用YOLOv5(6.0版本)时,运行export.py,尝试将pytorch训练pt模型转换成Tensorflow支持tflite模型,然而遇到报错: TensorFlow saved_model: export failure: can’t convert cuda:0 device type tensor to numpy. 对于此类问题,作者在issue中的统一回答是:新版本已解决了该问题,请使用新版本。
Test name:test_serialize_export_auto_functionalize_simple_cuda_float32 (__main__.TestHOPCUDA) Platforms for which to skip the test: linux, rocm, slow Disabled bypytorch-bot[bot] Within ~15 minutes,test_serialize_export_auto_functionalize_simple_cuda_float32 (__main__.TestHOPCUDA)will be di...
🐛 Describe the bug The exported model failed to do inference on cuda. import torch ep = torch.export.load('retina.pt2') gm = ep.module() gm(torch.rand(1, 3, 800, 1216)) # success gm = ep.module().cuda() gm(torch.rand(1, 3, 800, 1216).cud...
Test name: test_pre_dispatch_export_scan_simple_cuda_float32 (__main__.TestHOPCUDA) Platforms for which to skip the test: linux, slow Disabled by pytorch-bot[bot] Within ~15 minutes, test_pre_dispatch_export_scan_simple_cuda_float32 (__main__.TestHOPCUDA) will be disabled in PyTorch...
Tensors and Dynamic neural networks in Python with strong GPU acceleration - DISABLED test_retrace_export_auto_functionalize_simple_cuda_float32 (__main__.TestHOPCUDA) · pytorch/pytorch@0eb9c87
Tensors and Dynamic neural networks in Python with strong GPU acceleration - DISABLED test_retrace_export_auto_functionalize_simple_cuda_float32 (__main__.TestHOPCUDA) · pytorch/pytorch@9992084
PyTorch version: 1.11.0+cu113 Is debug build: False CUDA used to build PyTorch: 11.3 ROCM used to build PyTorch: N/A OS: Ubuntu 20.04.3 LTS (x86_64) GCC version: (Ubuntu 9.4.0-1ubuntu1~20.04.1) 9.4.0 Clang version: 10.0.0-4ubuntu1 CMake version: Could not collect Libc version...
To find relevant log snippets: Click on the workflow logs linked above Click on the Test step of the job so that it is expanded. Otherwise, the grepping will not work. Grep fortest_serialize_export_cond_simple_cuda_float32 There should be several instances run (as flaky tests are rerun ...
Tensors and Dynamic neural networks in Python with strong GPU acceleration - DISABLED test_retrace_export_while_loop_simple_cuda_float32 (__main__.TestHOPCUDA) · pytorch/pytorch@4aef5a1
Tensors and Dynamic neural networks in Python with strong GPU acceleration - DISABLED test_retrace_export_while_loop_simple_cuda_float32 (__main__.TestHOPCUDA) · pytorch/pytorch@65df26f