基于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”，修改完成后即可。
本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！