geo ai库本地运行测试的坑
geo ai库:GitHub - opengeos/geoai: GeoAI: Artificial Intelligence for Geospatial Data
坑1:Cannot find proj.db
安装pyproj库,通过设置环境变量 PROJ_LIB
指向包含 proj.db
的目录。
坑2:没法预测、训练
查看geoai库的依赖,发现torch、torchversion库是cpu版的,正常运行很慢。更改torch为gpu版,发现报错,没法运行。所以想到torch、torchversion、torchaudio都是相互关联的,于是按照pytorch官网指引先卸载然后重装,最后正常运行。搞定!!!
先卸载这两个库,然后按照自己的电脑环境重装这两个库。
pip uninstall torch
pip uninstall torchversion
测试管网样例:Train car detection - geoai
最后的运行结果,可以看到使用的是cuda: