Ubuntu16.04 配置ros+pytorch+cv2

Ubuntu16.04 配置ros+pytorch+cv2

最近安装这个环境,踩了很多坑,备份一下。

1、首先明确ros不一定完美支持python3,所以暂时先装在python2里。不过py2好像2021年停用。
2、如果想在python3搭环境,就确保pip3和所用的python3环境相同,可以通过pip3 -V查看。如果不相同的话,用软链接相连一下,这个网上有很多的教程,但注意要找到自己文件的位置。
3、要提前安装好驱动和cuda,N卡驱动可以在设置里面的其他驱动下载(需要重启),cuda我安装的是9.0版本。然后根据cuda版本去下载pytorch版本,网站:https://download.pytorch.org/whl/cu90/torch_stable.html
4、下载pytorch版本时注意,要提前确定好自己python版本可以安装的torch版本,我是ubuntu自带的python2.7,可以安装cp2.7mu,但是不能安装cp2.7m,最终选择了torch 0.4.1

下面说如果选择了python3环境:
1、ros虽然部分支持了python3的代码,需要在节点脚本前写好要运行的python版本(默认py2),即
#!/usr/bin/env python3
2、需要pip3 install一下缺少的包
3、如果遇到py3 import cv2出错,可以在环境中删掉ros对应的环境,或者将ros对应环境中的cv2.so更换一个名字 sudo mv cv2.so cv2_ros.so(如果更换名字,py2可能就import cv2失败,需要更改回来)
4、我python3 torch安装了1.1.0 cp35m

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值