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


版权声明:本文为xieshangxin原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。