我要评论0.前言
将Jetson TX2刷机完毕之后因为要用到pytorch，所以就在网上查找如何安装，但试了好多方法都失败了，然后找到了NVIDIA官网的安装教程，网址：nvidia官方。
1.步骤
（1）查看版本是否符合
上图是从所给网站上截下来的，可以看到支持安装JetPack4.2及以上版本的Nano,TX2,Xavier。而我的是安装有JetPack4.4 DP的TX2，是符合的。
（2）选择合适的Pytorch版本
从上面可以看到Pytorch v1.5.0和Pytorch v1.4.0都支持JetPack4.4 DP，因为之前安装了python3和pip3，所以我第一次选择了Pytorch v1.5.0，但后来出错了，如下：
显示Segmentation fault,这个错误的原因是torch的版本选择错误，所以我又把torch卸载，然后选择了Pytorch v1.4.0，之后就没有出现这种错误了。
（3）安装torch
按照上面的指示，在板子的终端依次输入Python3.6版本下的命令来安装torch，但这里很可能会出问题，wget https://nvidia.box.com/shared/static/ncgzus5o23uck9i5oth2n8n06k340l6k.whl -O torch-1.4.0-cp36-cp36m-linux_aarch64.whl这个命令可能执行不了，因为国内访问不了这个地址，这时候就要翻墙了，如果电脑上有翻墙软件直接翻墙即可访问下载。如果电脑上没有的话也可以用手机翻墙访问第（2）步中要下载的选项先下载下来然后放到板子的home里面，然后执行
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
pip3 install Cython
pip3 install numpy torch-1.4.0-cp36-cp36m-linux_aarch64.whl
即可。
如果顺利的话就可以安装好torch了。
（4）安装torchvision
安装torchvision的第二步要git一个版本的torchvision，而这个版本要参考上图，对应Pytorch v1.4选择了trochvision v0.5.0，命令如下：
sudo apt-get install libjpeg-dev zlib1g-dev
git clone --branch v0.5.0 https://github.com/pytorch/vision torchvision # see below for version of torchvision to download
cd torchvision
sudo python setup.py install
cd ../ # attempting to load torchvision from build dir will result in import error
顺利的话就成功了。
（4）测试
以上命令能正常输出就通过测试了。
本文地址：https://blog.csdn.net/comli_cn/article/details/107468083
您可能感兴趣的文章:
如您对本文有疑问或者有任何想说的，请 点击进行留言回复，万千网友为您解惑！
网友评论