Attempting pip install pytorch on jetson nano resulted in an error
# pip install torch
Collecting torch
Downloading https://files.pythonhosted.org/packages/f8/02/880b468bd382dc79896eaecbeb8ce95e9c4b99a24902874a2cef0b562cea/torch-0.1.2.post2.tar.gz (128kB)
100% |################################| 133kB 750kB/s
Complete output from command python setup.py egg_info:
running egg_info
creating pip-egg-info/torch.egg-info
writing requirements to pip-egg-info/torch.egg-info/requires.txt
writing pip-egg-info/torch.egg-info/PKG-INFO
writing top-level names to pip-egg-info/torch.egg-info/top_level.txt
writing dependency_links to pip-egg-info/torch.egg-info/dependency_links.txt
writing manifest file 'pip-egg-info/torch.egg-info/SOURCES.txt'
error: package directory 'torch/cuda' does not exist
----------------------------------------
Command "python setup.py egg_info" failed with error code 1 in /tmp/pip-build-7PRgOy/torch/
I found the article here but got an error again
gitclonehttp://github.com/dusty-nv/jetson-inference
cd tools
./install-pytorch.sh
# [jetson-inference] Package selection status: 0
[jetson-inference] Packages selected for download: 1
[jetson-inference] invalid PyTorch version selected: PyTorch python27
[jetson-inference] errors encountered during installation, exiting with code 1
[jetson-inference] to run this tool again, use the following commands:
$ cd <jetson-inference>/build
$ ./install-pytorch.sh
I was able to install it by referring to here.
# wget https://nvidia.box.com/shared/static/yhlmaie35hu8jv2xzvtxsh0rrpcu97yj.whl
# mv yhlmaie35hu8jv2xzvtxsh0rrpcu97yj.whl torch-1.4.0-cp27-cp27mu-linux_aarch64.whl
# pip install torch-1.4.0-cp27-cp27mu-linux_aarch64.whl
Processing ./torch-1.4.0-cp27-cp27mu-linux_aarch64.whl
Installing collected packages: torch
Successfully installed torch-1.4.0a0+7f73f1d
pytorch-for-jetson-version-1-7-0-now-available/72048 gitclonehttp://github.com/dusty-nv/jetson-inference
Recommended Posts