环境
TX2环境:
jetpack4.5.1
cuda10.2
需要安装的版本:
torch1.8.0
torchvision0.9.0
安装
torch
https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-10-now-available/72048
下载v1.8.0
sudo pip3 install torch-1.8.0-cp36-cp36m-linux_aarch64.whl +国内源
torchvision
https://github.com/pytorch/vision/tree/v0.9.0
copy to ubuntu
unzip
cd vision
sudo python3 setup.py install
mxnet (failed)
https://github.com/apache/incubator-mxnet/issues/19234
scipy
sudo apt-get install libblas-dev liblapack-dev libatlas-base-dev gfortran
(TX2好像默认有scipy,更新即可)
sudo pip3 install --upgrade scipy -i https://pypi.tuna.tsinghua.edu.cn/simple
matplotlib
import matplotlib
matplotlib.use(“agg”)
import matplot.pyplot as plt
否则会报GTK2.X\GTK3.X问题