Hi, i am trying to optimise my resnet18 model using tensor2trt. But its failing with below error. Im using the existing road following repo
This is happening when i am trying to convert model to trt with torch.save(). I haven’t modified any code in the givin repo.
Github: jetracer/notebooks/optimize_model.ipynb at master · NVIDIA-AI-IOT/jetracer · GitHub
Cuda 10.2
Cudnn 8
Torch 1.6.0
Jetpack 4.5
Hi,
Do you use the same environment shared in the Jetracer?
https://github.com/NVIDIA-AI-IOT/jetracer/tree/master
6/01/2021 - JetPack 4.5.1 based image is updated. It is pre-configured for JetRacer. Detail here .
Thanks.
We followed the instructions provided in the github.!! What you mean by environment is it jetpack version?? Can you pls help me understand
Hi,
Yes, do you use the image shared in the link below?
# Software Setup
> If you have built a 3rd-pary Jetracer kit (other than our DIY vehicles based on Latrax Rally or Tamiya TT-02), the following method (and the SD card images) may not work. <br> **Please check the manufacture's instruction on its kit's specific software setup.**
After finishing these steps, you'll be all set to program JetRacer. Once you're finished, run through the [examples](examples.md).
## Step 1 - Flash micro SD card
1. Download a JetCard SD card image listed in below table onto a Windows, Linux or Mac *desktop machine*
2. Insert a 32GB+ SD card into the desktop machine
3. Using [Etcher](https://www.balena.io/etcher/) select the downloaded zip file and flash it onto the SD card
4. Remove the SD card from the desktop machine
### Latest Release (** **but not yet fully verified** ** )
> Please note, the password for the pre-built SD card is ``jetson``
| Platform | Board revision | JetPack Version | Download | MD5 Checksum |
| -------- | -------------- | --------------- | -------- |------------- |
| Jetson Nano (4GB) | `A02` and `B01` | 4.5.1 | [jetcard_nano-4gb-jp451.zip](https://drive.google.com/file/d/1aPbzQ0_Uja0jVD48oZUAuYYAz10JgbZu/view?usp=sharing) | 3195c91e6069c0418ec3c9736d130d01 |
This file has been truncated. show original
Thanks.
Yes same image is used. It worked yesterday but i dono why it dint work for the past 5 weeks. So this happens only cos of the image? Or memory issue??
Hi,
Not sure but do you mean it works now?
Thanks.
Yes as you mentioned earlier about the image, i made sure to download it by myself and flash it. Another thing is i dint train the model as i had the pth file already… so i directly went for converting it into trt and it worked so not sure about the cause… will try to replicate and see
Thanks for the guidance
system
Closed
August 28, 2024, 6:50am
10
This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.