TX2：安装pytorch(Jetpack3.3+cuda9.0+cudnn7)

Jetpack3.3刷机+查看cuda+cudnn

刷机过程参考：https://blog.csdn.net/NSSC_K/article/details/106092499
本人使用虚拟机Ubuntu16.04来对TX2进行刷机，虚拟机的网络连接一定要设置桥接模式，不然后面刷机会卡住。
刷机成功后，在TX2查看cuda版本：

nvcc -v

如果报错：

-bash: nvcc: command not found

解决：

sudo getdit ~/.bashrc
#最后加
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64：
export PATH=$PATH:/usr/local/cuda-9.0/bin

查看cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

本人发现找不到cudnn.h文件，可能刷机的时候只装好了cuda,需要自己另行安装cudnn。
可以从官网或者清华镜像下载所需的cudnn版本。由于官网速度太慢，还需要注册，本人直接从清华镜像里下载 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

conda install --use-local cudnn--7.6.5-cuda9.0_0.tar.bz2
mv /usr/include/cudnn.h /usr/local/cuda/include/
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

安装conda

TX2无法使用anaconda，但可以用archiconda。直接从https://github.com/Archiconda/build-tools/releases这里下载相应的版本

sh ./Archiconda3-0.2.2-Linux-aarch64.sh

安装pycharm

https://www.jetbrains.com/pycharm/download/#section=linux里直接下载
解压后进入pycharm文件夹中的bin文件夹下，打开终端

sh ./pycharm.sh

打开pycharm后，设置里Create Desktop Entry

安装pytorch(cuda9.0+cudnn7)

conda create –n pytorch python=3.6
source activate pytorch

这里本人用源码安装Pytorch

git clone --recursive --branch v1.0.0 http://github.com/pytorch/pytorch

如果报错可尝试将http改为git