Jetson Xavier NX install Pytorch

由于NX使用了CUDA10.2,而不是10.0,导致当下载一些软件包的时候会出现不兼容的问题,比如之前在Jetson Nano可以运行的torch1.1,在NX上就会报错 Import Error: libcudart.so.10.0: cannot open shared object file: No such file or directory

解决办法:更换软件包版本
下载torchvisionv1.5就可以与cuda10.2兼容了

torch-1.5.0-cp36-cp36m-linux_aarch64安装包

链接:https://pan.baidu.com/s/1nkZ9gL5VfAMoHP-cyY7l7g
提取码:et98

之后再通过 pip install torchvision 会自动安装0.2.2版本的torchvision

至此,PyTorch安装成功

你可能感兴趣的:(nvidia,cuda,ai)