nvidia-jetsion tx2 上安装torch torchvision 遇到的一些坑

1、直接 pip install torch ,直接安装了torch v1.8.1，但是不能使用GPU

torch.cuda.is_available() 结果为false

2、安装torch v1.8.0.whl【wget下载】，然后 pip install torchvision，安装ttorchvision0.9.1成功，并且导入成功，但是在函数调用时报错。

3、git clone  torchvision 重新安装0.9.0,源码安装成功【setup.py】，但是 查看torchvision.__version__，依然是0.9.1。 卸载它之后，查看torchvision.__version__ ， 0.9.0

4、重新运行程序，没有报错

 

注：torch 和torchvision 严格匹配。

pip install torchvision==0.9.0没有成功，找不到，必需要git clone.

 

参考 https://blog.csdn.net/weixin_43947712/article/details/115530913