基于RTX30显卡的ArcGIS Pro2.8深度学习环境配置
问题描述
基于我不知道的原因,ArcGIS Pro2.8并不支持RTX30显卡,在深度学习模型训练中会出现模型精度极低,loss等值为nan的情况,具体可看ESRI讨论贴:https://github.com/Esri/deep-learning-frameworks/issues/17
帖中总结出的问题具体如下:
解决思路
进入正题,配置argis pro2.8深度学习环境以支持RTX30显卡的思路是:
1、安装CUDA11和对应cudnn;
2、arcgis pro2.8深度学习环境库包替换。
1、安装CUDA11+cudnn
第一步:检查电脑的显卡类型
第二步:下载cuda11安装包并安装
博主的显卡是11.4,但是安装了11.1的cuda,这是基于arcgis pro深度学习库包版本考虑,最后应验证cuda安装成功。
CUDA11.1下载地址:https://developer.nvidia.com/cuda-11.1.1-download-archive
第三步:安装对应版本的cudnn并配置环境变量
cudnn下载地址:https://developer.nvidia.com/zh-cn/cudnn
以上参照博客配置即可:https://blog.csdn.net/A496608119/article/details/112855153
配置ArcGIS Pro2.8深度学习环境
第一步:配置深度学习环境arcgispro-py3
安装参照ESRI Github:https://github.com/Esri/deep-learning-frameworks/blob/master/README.md?rmedium=links_esri_com_b_d&rsource=https%3A%2F%2Flinks.esri.com%2Fdeep-learning-framework-install
下载ArcGIS Pro2.8深度学习环境配置程序:
下载完成后运行MSI文件,运行完成后,ArcGIS Pro2.8便可在自带arcgispro-py3环境中执行深度学习工具。
运行完成后打开ArcGIS Pro2.8,执行训练模型工具,可执行成功即代表环境配置成功。
第二步:克隆arcgispro-py3环境到deepleanring
打开ArcGIS Pro的python交互式终端
输入命令:conda create -n deeplearning --clone arcgispro-py3
完成自有环境的克隆后,输入命令:activate deepleanring
激活新环境后,将在此环境中进行库包的替换。
第三步:替换deeplearning环境中的库包
先贴上我配置完成的库包列表:
(deeplearning) C:\WINDOWS\system32>conda list
替换的库包应下载ESRI提供的库包,地址:https://anaconda.org/esri/repo?page=1
主要替换pytorch所关联到的库包,所有替换的库包可以直接下载tar.bz2文件,库包不进行升级或降级,核心库包pytorch仍旧采用1.4.0版本,ESRI提供了基于cuda11.2+cudnn8.1的pytorch:
ESRI提供的下载源没有的库包,可以从清华源下载:https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
下载好的库包采用conda命令安装:conda install ********.tar.bz2(***为库包文件本地路径)。
最后输入:proswap deeplearning,将ArcGIS Pro2.8的python环境替换为新环境。
第四步:修改ArcGIS Pro的小BUG
训练模型时会GBK错误,需要修改
C:\Program Files\ArcGIS\Pro\bin\Python\envs\deeplearning\Lib\site-packages\arcgis\learn\_data.py中的代码,在脚本中with open() as f中添加encoding=“utf-8”,修改完成后即可。
更多推荐
所有评论(0)