GraspNet-baseline复现----Linux-Ubuntu

1.基本环境

  • Ubuntu 20.04
  • Cuda 11.0 、cuDNN 80.0
  • Python 3.7.16
  • PyTorch 1.7.0

2.环境配置

PyTorch的版本对Cuda和Python的版本都有依赖,所以基本步骤是 确定需要安装的PyTorch版本 —> 通过 网站 确定对应的cuda版本和python版本 —> 创建虚拟环境配置环境。

# 1.创建虚拟环境,配置python
conda create -n 虚拟环境名 python=3.7

# 2.配置虚拟环境的cuda、cuDNN和PyTorch
根据这篇博客:https://blog.csdn.net/qq_42194665/article/details/130028584

# 3.检查虚拟环境中的cuda、cuDNN、PyTorch是否配置好(上面这篇博客)


3.预训练模型运行

根据github上的来。

注意Tolerance Label可以运行command_generate_tolerance_label.sh生成,或者直接从云盘里下载后解压,放到dataset文件夹。

下载的预训练权重文件路径要记得,打开command_demo.sh文件,将 ”logs/log_kn/checkpoint.tar“ 改为自己的权重文件路径。

执行示例demo脚本

bash command_demo.sh

结果如下:
在这里插入图片描述

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值