描述:我安装的是torchvision0.8.0,因为在这种嵌入式设备中在线安装的torch和torchvision库不能正确的使用GPU,只能使用cpu,所以我们需要采用离线安装的方式。这是我踩过的一些坑,分享给大家。
首先需要将vision安装包克隆下来,然后按照我这个流程安装就行,注意torch、torchvision和python版本对应关系。
git clone -b v0.8.1 https://github.com/pytorch/vision.git vision-0.8.1
安装:
cd vision-0.8.1
sudo /home/nvidia/mambaforge/envs/ultra-fast-lane/bin/python3.6 setup.py install
注意:务必指定路径,且sudo
错误:非法指令
解决方法:降低numpy版本,我用的numpy=1.19.3,成功