在使用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_pre_dispatch_export_invoke_subgraph_simple_cuda_float32 (__main__.TestHOPCUDA) Platforms for which to skip the test: linux, rocm, slow Disabled by pytorch-bot[bot] Within ~15 minutes, test_pre_dispatch_export_invoke_subgraph_simple_cuda_float32 (__main__.TestHOPCUDA) wi...
🐛 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...
Tensors and Dynamic neural networks in Python with strong GPU acceleration - `torch._export.aot_compile` CUDA version not compatible with C++ ABI · pytorch/pytorch@a1ba8e6
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...
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT - TensorRT/examples/dynamo/torch_export_cudagraphs.py at 262c87dab545973fd4a3a6b26a22df11a5fe841f · pytorch/TensorRT
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 ...
linux-jammy-cuda11.8-cudnn9-py3.9-clang12 / build Node.js 16 actions are deprecated. Please update the following actions to use Node.js 20: pytorch/test-infra/.github/actions/setup-ssh@main, malfet/checkout@silent-checkout, seemethere/upload-artifact-s3@v5. For more information see: ht...
Tensors and Dynamic neural networks in Python with strong GPU acceleration - DISABLED test_pre_dispatch_export_auto_functionalize_simple_cuda_float32 (__main__.TestHOPCUDA) · pytorch/pytorch@15b5a0b
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