Hi, I am following this documentation on re-training SSD model - jetson-inference/pytorch-ssd.md at master · dusty-nv/jetson-inference · GitHub . I have downloaded images that I want and I am running train_ssd.py, but it runs into cuda error as below. THCudaCheck FAIL file=/media/nvidia/WD_BLUE_2.5…

Hi, This is a similar issue of this topic . It can be fixed with this change: https://gist.github.com/dusty-nv/ce51796085178e1f38e3c6a1663a93a1#file-pytorch-1-6-rc2-jetpack-4-4-ga-patch Not sure which package is included in the SDK. Would you mind to give ab…

Cuda runtime error while re-training SSD

Robotics & Edge Computing Jetson & Embedded Systems Jetson Nano

AastaLLL October 5, 2020, 6:07am 3

Him

May I know how do you install the pyTorch package?
It’s recommended to install the version shared in the below’s comment:

Thanks.

Topic		Replies	Views
Jetson Nano Torch 1.6.0 PyTorch Vision v0.7.0-rc2 Runtime Error Jetson Nano pytorch	4	1436	October 18, 2021
Getting error as Cuda Runtime (invalid argument) Jetson Nano cuda	12	1723	September 25, 2023
Nano PyTorch 1.2.0 wheel Jetson Nano	8	1433	October 18, 2021
Problem with Re-training SSD-Mobilenet Jetson Nano cuda , tensorflow , jetson-inference , python	2	809	March 3, 2022
Train_ssd.py dosen't work with pascal voc dataset Jetson Nano ai-training	5	1125	February 9, 2022
Error during re-training SSD-Mobilenet using Jetson Nano 2GB Jetson Nano jetson-inference	3	16	April 16, 2025
Pytorch installation error General Topics and Other SDKs tensorflow , jetson-inference	10	1233	December 14, 2023
Cannot install Pytorch 2.x with CUDA support Jetson Orin Nano cuda , pytorch	9	817	September 2, 2024
Problem with mobilenet-v1-ssd-mp-0_675.pth when re-training SSD-MOBILENET Jetson Nano tensorrt , cuda , jetson-inference , python	2	1465	March 3, 2022
Pytorch support Jetson Nano	31	4536	October 18, 2021

Cuda runtime error while re-training SSD

Related topics