PyTorch with CUDA 10.2 using Python3.8 in Jetson Nano (Jetpack 4.6.2)

[bawinsh04]

Continuing the discussion from Jetson Nano Pytorch Wheel files for latest Python 3.8/3.9 versions with CUDA 10.2 support:

@dusty_nv

I’ve tried to compile the pytorch-1.8 (the master branch on Dec 12, 2020) on JetPack 4.3 (cuda-10.2, cudnn-8) image (ubuntu 18.04) using the gcc-7.5, gcc-8.4, gcc-9.3 and gcc-10.1. Finally, only the gcc-9.3 can be used to compile the code and generate the final wheel. The generated wheels can be found in https://icedrive.net/1/76FFGZiyDO.

I was build a file wheel for PyTorch 1.7, Python 3.8 by Jetson Nano. @dusty_nv Please verify and confirm for otherspeople. Download at: torch-1.7.0a0-cp38-cp38-linux_aarch64.whl - Google Drive

I’m using Jetson Nano with Jetpack 4.6.2 and have Python3.8 installed in it.
For my defect detection Yolo Model i need pytorch to be installed with CUDA.
For installation of pytorch with cuda support using python3.8 which of these 2 files should i use ??

[AastaLLL]

Hi,

For JetPack 4, we only provide the package built with Python 3.6.

jp4/cu102 index

But you can try to build it from the source on your own.

Thanks.