I have followed GET STARTED Start Locally | PyTorch reference, installed all prerequsites, but below command returned an error. I’ve been trying to install torch and torchvision on Jetson Nano in my virtual envaironment (torch). Please help.
(torch) $ pip3 install torch torchvision
…
ERROR: Command errored out with exit status 1: python setup.py egg_info Check the logs for full command output.
You can build it from source with the python you preferred.
You did not answer my MAGMA question. Your suggestion won’t work unless NVIDIA/CUDA provided full support for Python 3.7+. Ok. I’ll raise this question about Jetson Nano Python 3.7+ support again, when Google CoLab (or any NVIDIA partner) has migrated from Python 3.6 to 3.7+.
Why do you need python 3.7? Nvidia uses the same version as Ubuntu 18.04, and I mean the exact same version because Canonical builds it. Pytorch seems to only require 3.5 anyway and recommend version is 3.6.
As Nvidia rep said, if you build your own for python 3.7 it should work. It’s not going to be any faster with 3.7, however, and you’re likely only going to run into trouble using a version that isn’t recommend by the package author.
Why do you need python 3.7?
There are some latest packages which are not compatible with 3.6 (can’t disclose details). I have faced this issue and tryed to resolve the issue in Jetson Nano Python 3.7 virtualenv. I have failed to compile torch/torchvision due to missing Python 3.7 support for MAGMA (LAPACK support for the GPU) for Jetson Nano. Probaly some other “torch/torchvision install from source” prerequsites are missing for Jetson Nano as well.
CUDA toolkit is a C++ based library. Suppose it won’t stop you to use python3.7.
It looks like the issue occurs from LAPACK? Is it correct?
You may need to build lots of libraries from source if you prefer a non-default python version.
I beleive the torch and torchvision Python 3.7.4 issue on Jetson Nano is not related to CUDA. Most likely the issue is with LAPACK (and MAGMA). Will try to follow your advise… When do you plan official support of Python 3.7+ on Jetson Nano?
Hi, You are right, but you can install latest Python >3.6 in docker container with all CUDA support based on CUDA verion installd on Jetson Nano host. If you have Jetson Nano image without Python 2.x and with Python >3.8.0, please let me know.