search

opendr-eu / opendr

A modular, open and non-proprietary toolkit for core robotic functionalities by harnessing deep learning
Apache License 2.0
618 stars 95 forks source link

Pytorch installation on Jetson #218

Closed iliiliiliili closed 2 years ago

iliiliiliili commented 2 years ago

Pytorch for Jetson devices with GPU support cannot be installed directly and has to be installed using provided pip wheels. These wheels are available for Python 3.6 only and cannot be installed via pip install torch==1.7.1. How should it be incorporated into Jetson installation? I believe, it is best to have separate install scripts for PC and Jetson devices.

passalis commented 2 years ago

Did you use the install_tx2.sh script? I think it downloads one of these wheels.

vniclas commented 2 years ago

@passalis I am a bit confused concerning the installation on Jetson devices. Is the script linked in your reply supposed to be used for all Jetson models or just for TX2?

passalis commented 2 years ago

The script (I fixed the name in my previous answer) was developed for TX-2. Also, I am not sure that it supports all the tools, so each partner should check its own tools (#195 is open). This is part of the preparation for D8.1. It can be also used as a template to develop similar scripts for other jetsons.

Pavlos-Tosidis commented 2 years ago

There are instructions in #195 on how to run the script on Xavier AGX and NX. As @passalis mentioned, some tools are not supported due to some dependencies missing, so each partner should check its own tools.

Pavlos-Tosidis commented 2 years ago

Closing this, as #195 address this.