TX2上安装pytorch

由于TX2是arm架构,无法直接通过官方的方法——pip或conda等安装,需要自行编译github上的软件包才能使用。

注:本文略过TX2上的CUDA,Cudnn以及opencv的安装,这些部分的安装大家可以自行百度。

现在正式开始编译pytorch吧:

1.打开终端,输入以下命令下载git包

git clone https://github.com/pytorch/pytorch.git

下载完后,大家会看到在home下有了pytorch文件夹。

2.切分版本

cd到home下的pytorch文件夹,进行版本切分,我要使用pytorch0.3.1版本,所以依次输入以下命令进行切分:

cd '/home/nvidia/pytorch'

git checkout -b v0.3.1 v0.3.1

3.更新submodule

此步不进行,在install的时候会报Could not find /home/nvidia/pytorch/torch/lib/gloo/CMakeLists.txt错误

在pytorch文件夹路径下,在终端输入以下命令进行更新:

git submodule update --init

4.正式编译安装

还是在当前pytorch文件夹路径下,输入以下命令进行编译安装:

sudo python setup.py install

切记一定要带上sudo,不然编译很久很久要完成后,你会发现报Permission denied的错误。

(编译时间很长,大家请耐心等待,睡一觉就好了)

5.测试

在python下输入import torch,没有显示错误则说明编译安装成功,如下图:

  • 3
    点赞
  • 24
    收藏
    觉得还不错? 一键收藏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值