nvidia apex xavier 安装torch1.7.0

apex xavier不能直接pip安装torch和torchvision库,否则会导致gpu无法使用,需手动下载安装
此处给出两个连接
1邮箱中转站
2百度网盘链接:
https://pan.baidu.com/s/19-HtGDomw5PWteym7MaG0A提取码: tx8i 复制这段内容后打开百度网盘手机App,操作更方便哦

将安装包下载后

pip install torch-1.7.0-cp36-cp36m-linux_aarch64.whl

安装torchvision参考我的另外一篇文章:
一定要使用sudo获取权限安装的制定路径里,不然可能会安装到base环境中。其余的库就可以conda或者pip指令安装了。就可以跑yolov5等pytorch代码

你可能感兴趣的:(xavier,python,开发语言,后端)