返回顶部

[天文摄影] 【PI后期4】利用显卡GPU加速Starnet++,速度快了5倍以上

[复制链接]
田麦Lv.16 显示全部楼层 发表于 2022-5-20 10:45 |阅读模式 打印 上一主题 下一主题

马上注册,结交更多好友,享用更多功能,让你轻松玩转社区。

您需要 登录 才可以下载或查看,没有账号?加入牧夫(注明天文爱好者)

x
本帖最后由 田麦 于 2022-5-28 21:36 编辑

Starnet++是一个免费的优秀去星软件。可惜运行的速度太慢了。怎样才能使它运行更快一些呢?方法是让显卡的GPU参与运算。借助GPU的算力,可以比单独使用CPU运算时的运行速度快了5倍以上。

本文介绍的GPU加速方法有一个前提:仅适用于64位的WINDOWS和nVidia GPU显卡的特定型号:(可官网查询支持CUDA的GPU显卡:https://developer.nvidia.com/cuda-gpus,需要列表中的Compute Capability达到或超过7.0以上),以及 Pixinsight 1.8.8以上。

需用到的软件:
nVidia CUDA 10.1 (Update 2)
nVidia cuDNN v7.6.5, for CUDA 10.1
libtensorflow-gpu 2.3.0
网盘下载链接:https://pan.baidu.com/s/1T5r3dnEXOC7H5jQssrPuRg
提取码:0000

(新增软件可以安装在Pixinsight同一个硬盘中,也可以安装在其它硬盘中,下面以X: 代替,请留意替换为具体盘符。安装过程中因为需要下载库数据,故要联网)

一、替换 tensorflow.dll
在libtensorflow-gpu-windows-x86_64-2.3.0\lib文件夹中,将tensorflow.dll 复制粘贴到“X:\Program Files\PixInsight\bin”下,覆盖现有文件。
(为保险起见,可以事先把原来的tensorflow.dll 改名为tensorflow_old.dll )

二、安装CUDA
1、运行 cuda_10.1.243_win10_network.exe
2、选择“自定义(高级)”
1.png

3、组件中仅仅勾选CUDA-Runtime-Libraries。请仔细检查,确保除此之外的其它所有内容都取消勾选。然后按下一步,直至安装完成。
2.jpg


三、安装 cuDNN
将cudnn-v7.6.5.32_cuda10.1-windows10-x64\cuda\中的子文件夹“bin”和“lib”复制粘贴到“X:\Program Files\NVIDIA GPU Computing Toolit\CUDA\V10.1”中。
3.png

四、编辑环境变量
您将需要在 Windows 中设置 2 个环境变量,以便一切正常运行。
1、在Win开始菜单的搜索栏中搜索“编辑”,选择“编辑系统环境变量(控制面板)”,并点击后进入系统属性窗口。
4.jpg

2、在“系统属性”窗口中点击其中的“环境变量N”
5.jpg

3、在弹窗下部的“系统变量S”窗口下,单击“新建”并输入“ TF_FORCE_GPU_ALLOW_GROWTH ”作为变量名和“ true ”作为变量值,按 确认。
6.jpg

4、在弹窗下部的“系统变量S”列表中找到名为“Path”的变量,点选它并单击“编辑”。
7.jpg

5、点击“新建”,新建 X:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin”后按“确定”退出。
8.jpg
到此,软件全部安装完毕。(安装后最好重启一次电脑)

五、验证
打开 Windows 任务管理器,转到“性能”选项卡并选择 GPU 0,点击左下方图表上方的Video Encode,在下拉菜单中找到CUDA并选择它。
在Pixinsight中运行Starnet++,您会看到随着进程的开始,GPU曲线急剧上升,还有Starnet++运行速度的大大加快。
如果看不到这样的结果,很多时候是有关环境变量的问题。请详细检查环境变量的设定是否正确。
9.jpg

利用GPU加速后,用Starnet++(256)处理一个 6248 X 4176的彩色图像,用时仅29.523秒!一般地说来,采用GPU加速,Starnet++可以加速5倍以上。你也不妨试试。
10.jpg

(本文编译于DARKARCHON的《[UPDATED FOR PIXINSIGHT 1.8.8-6] PIXINSIGHT, STARNET++ AND CUDA – GOTTA GO FAST》)


===================================

                                   目       录
—————————————————————————

——————————————————————————



—————————————————————————






精彩评论6

ksolLv.4 来自手机 显示全部楼层 发表于 2022-5-20 16:04
回复 支持 反对

使用道具 举报

tauxLv.3 来自手机 显示全部楼层 发表于 2022-5-21 08:05
谢谢田麦老师。
回复 支持 反对

使用道具 举报

tauxLv.3 来自手机 显示全部楼层 发表于 2022-5-21 08:05
谢谢田麦老师。
回复 支持 反对

使用道具 举报

RaylppLv.10 来自手机 显示全部楼层 发表于 2022-5-23 12:53
果然是运算能力要7以上,我6点多算力的显卡运行起来比cpu快不了多少

                               
登录/注册后可看大图
回复 支持 反对

使用道具 举报

lanneret_skyLv.2 来自手机 显示全部楼层 发表于 2022-5-26 06:33
回复 支持 反对

使用道具 举报

xlongLv.10 显示全部楼层 发表于 2022-5-26 22:58
谢谢,用上了,加速很明显
回复 支持 反对

使用道具 举报

本版积分规则

爱牧夫天文科技(大连)有限公司是国家批准成立的正规光学科技仪器研发企业,还是全国着名的天文科技文化传播公司,旗下拥有中国最大最早的天文论坛网站——牧夫天文论坛,拥有会员近五万,日访问量超过22万,其背后有中国国家天文台、《中国国家地理》杂志社、世界天文权威杂志美国的《天空和望远镜》杂志等机构的支持。
  • 官方手机版

  • 微信公众号

  • ˊ
  • Powered by Discuz! X3.4 | Copyright © 2001-2021, 愛牧夫天文科技(大连)有限公司. |
  • | 营业执照 | |牧夫天文|辽ICP备19018387号 |辽公网安备 21021102000967号 QQ