How to install Pytorch 1.8.0 with cuda support and torchvision

Autonomous Machines Jetson & Embedded Systems Jetson TX2
1

Good day, I’m using jetson tx2 to use the yolov8, and the only problem now is that I can’t run my yolov8 model in gpu because the pytorch is not cuda enabled. I’m posting here since I can’t find in forums on how to install pytorch with cuda support.

3

Hi,

Please find the info on below topic:

JetPack 4
PyTorch v1.8.0

Thanks.

4

I’m sorry I didn’t put the python version on my post but I want to install it on python 3.8.

5

Hi,

We only provide python 3.6 prebuilt for JetPack 4.
For other versions, please build it from the source.

Thanks.

7

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.