当安装遥感图像深度学习所需要的包(如 tensorflow 和 gdal)时,需要遵循特定的安装顺序。由于 gdal 对包的依赖兼容性比较差,而 gdal 和 tensorflow 又有共同依赖的包(例如 numpy、h5py 等等),因此安装顺序很重要。如果先安装了 tensorflow 包再安装 gdal 包,共同依赖的包可能只安装了一次,这些共同包可以兼容 tensorflow,但无法兼容 gdal。这将导致在调用时显示无法找到 gdal 文件。
在遇到报错时,最直接的方法是将两个包都先删除,然后先安装 gdal 再安装 tensorflow。卸载 gdal 包时,要使用 conda 卸载:
conda uninstall gdal
卸载 tensorflow 包时,可以直接使用 pip 卸载:
pip uninstall tensorflow
要安装 gdal,要使用 conda 安装:
conda install gdal
要安装 tensorflow,则可以使用本地已下载的 .whl 文件安装,假设您要安装的版本是 tensorflow-2.6.0:
pip install tensorflow-2.6.0-cp39-cp39-win_amd64.whl
可以从 https://pypi.org/project/tensorflow/2.6.0/#files 下载相应的文件。注意,具体的 tensorflow 版本要和自己安装的 CUDA 和 cuDNN 版本对应。例如,如果您的版本是 python3.9 + CUDA11.2 + cuDNN8.1 + tensorflow2.6.0,则应该选择相应的版本进行安装。
同时,注意 CUDA 和 cuDNN 以及 tensorflow 版本之间有严格的对应关系。如果不对应,则可能会成功安装,但无法使用 GPU。因此,务必保证这些版本要严格对应。