tx2 cuda9

nvcc --version

this project provide a complied wheel file of pytorch1.0 with cuda support for nvidia jetson tx2.

the environment base is jetpack 3.3 and python3.5

cuda9.0 and cudnn7.1.5 was default setting from jetpack 3.3

download link:https://drive.google.com/open?id=1ESslprUDCBbXu69SE5EwkHZuoBPD3tgS

wget http://developer.download.nvidia.com/devzone/devcenter/mobile/jetpack_l4t/013/linux-x64/cuda-repo-l4t-8-0-local_8.0.84-1_arm64.deb

$ sudo dpkg -i cuda-repo-l4t-8-0-local_8.0.84-1_arm64.deb

cuda-repo-l4t-9-0-local_9.0.252-1_arm64.deb

$ sudo apt-get update

$ sudo apt-get install cuda-toolkit-8-0

sudo apt-get install cuda-toolkit-9-0
sudo apt-get remove cuda-toolkit-9-0

https://blog.csdn.net/zsfcg/article/details/84099869

python setup.py install

sudo python setup.py clean

ls /usr/local/cuda-8.0/lib64
and try install via:

CUDA_HOME="/usr/local/cuda-9.0" python setup.py install