本帖最后由 田麦 于 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、选择“自定义(高级)”
3、组件中仅仅勾选CUDA-Runtime-Libraries。请仔细检查,确保除此之外的其它所有内容都取消勾选。然后按下一步,直至安装完成。
三、安装 cuDNN
将cudnn-v7.6.5.32_cuda10.1-windows10-x64\cuda\中的子文件夹“bin”和“lib”复制粘贴到“X:\Program Files\NVIDIA GPU Computing Toolit\CUDA\V10.1”中。
四、编辑环境变量
您将需要在 Windows 中设置 2 个环境变量,以便一切正常运行。
1、在Win开始菜单的搜索栏中搜索“编辑”,选择“编辑系统环境变量(控制面板)”,并点击后进入系统属性窗口。
2、在“系统属性”窗口中点击其中的“环境变量N”
3、在弹窗下部的“系统变量S”窗口下,单击“新建”并输入“ TF_FORCE_GPU_ALLOW_GROWTH ”作为变量名和“ true ”作为变量值,按 确认。
4、在弹窗下部的“系统变量S”列表中找到名为“Path”的变量,点选它并单击“编辑”。
5、点击“新建”,新建 X:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin”后按“确定”退出。
到此,软件全部安装完毕。(安装后最好重启一次电脑)
五、验证
打开 Windows 任务管理器,转到“性能”选项卡并选择 GPU 0,点击左下方图表上方的Video Encode,在下拉菜单中找到CUDA并选择它。
在Pixinsight中运行Starnet++,您会看到随着进程的开始,GPU曲线急剧上升,还有Starnet++运行速度的大大加快。
如果看不到这样的结果,很多时候是有关环境变量的问题。请详细检查环境变量的设定是否正确。
利用GPU加速后,用Starnet++(256)处理一个 6248 X 4176的彩色图像,用时仅29.523秒!一般地说来,采用GPU加速,Starnet++可以加速5倍以上。你也不妨试试。
(本文编译于DARKARCHON的《[UPDATED FOR PIXINSIGHT 1.8.8-6] PIXINSIGHT, STARNET++ AND CUDA – GOTTA GO FAST》)
===================================
目 录—————————————————————————
——————————————————————————
—————————————————————————
|