Jetson TX1 installation pytoch

General process

1. Reshipment system

Reinstall the system first , Avoid unnecessary troubles caused by chaotic old systems .

Install dependency packages

sudo apt-get install python3-pip cmake
pip3 install numpy pyyaml mkl mkl-include setuptools cffi typing
#  It is also recommended to install other packages in the major tutorials , I don't know if it's useful .

download Pytorch Source code

#  If you want to install the historical version , You can download the source code of the corresponding version , perhaps git checkout
git clone --recursive http://github.com/pytorch/pytorch
cd pytorch

Install what is required in the source code Python Dependency package

sudo pip3 install -r requirements.txt

install

#  Direct installation 
sudo python3 setup.py install

#  Generate whl
python3 setup.py bdist_wheel

A reference link for installation is attached ：

1. https://gist.github.com/dusty-nv/ef2b372301c00c0a9d3203e42fd83426
2. https://github.com/andrewadare/jetson-tx2-pytorch
3. PyTorch for Jetson Nano - NVIDIA Developer Forums
4. https://github.com/dusty-nv/jetson-reinforcement/blob/master/CMakePreBuild.sh

Possible problems ：

1. Out of memory

#  The displayed error message is not saved 
#  It seems to be some internal error  

terms of settlement ： Add swap
A reference link for adding partitions is attached ：
https://help.ubuntu.com/community/SwapFaq#How_do_I_add_or_modify_a_swap_partition.3F
https://www.jetsonhacks.com/2016/12/21/jetson-tx1-swap-file-and-development-preparation/

2. NCCL Compiler error

Makefile:68: recipe for target '/home/ubuntu/software/pytorch/build/nccl/obj/collectives/device/devlink.o' failed
make[5]: *** [/home/ubuntu/software/pytorch/build/nccl/obj/collectives/device/devlink.o] Error 255
Makefile:44: recipe for target '/home/ubuntu/software/pytorch/build/nccl/obj/collectives/device/colldevice.a' failed
make[4]: *** [/home/ubuntu/software/pytorch/build/nccl/obj/collectives/device/colldevice.a] Error 2
Makefile:25: recipe for target 'src.build' failed
make[3]: *** [src.build] Error 2
CMakeFiles/nccl_external.dir/build.make:110: recipe for target 'nccl_external-prefix/src/nccl_external-stamp/nccl_external-build' failed
make[2]: *** [nccl_external-prefix/src/nccl_external-stamp/nccl_external-build] Error 2
CMakeFiles/Makefile2:67: recipe for target 'CMakeFiles/nccl_external.dir/all' failed
make[1]: *** [CMakeFiles/nccl_external.dir/all] Error 2

because TX1 By default, it is not installed NCCL, Therefore, in order to avoid error reporting , But for CMakeLists.txt and setup.py Make the following changes ：

# append to setup.py
USE_NCCL = False
# USE_DISTRIBUTED = False

# change the option in the CMakeLists.txt
option(USE_NCCL "Use NCCL" OFF)

After modifying the parameters , Suggest clean once ！！！