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