在Windows和Linux环境下都跑通了,大约用了一周的时间,大部分原因都是版本不一致造成的:
1、本机的 CUDA 与 paddlepaddle-gpu 的版本准确匹配
step 1:如果 paddlepaddle-gpu 只有 11.8,那么本机就得安装 CUDA11.8 及其相应CUDNN
在如下链接找到对应版本的 paddlepaddle-gpu: www.paddlepaddle.org.cn/install/qui…
step 2:如果我们机子上有多个 cuda,记得在运行的 conda 环境下,
Windows系统下:修改环境变量,Path 将要用的版本放在最前面;CUDA_PATH 修改为要用的版本。
Linux系统下:用脚本switch-cuda.sh(网上搜得到)转换一下到匹配版本:
source switch-cuda.sh 11.8
使用 nvcc -V
对版本验证是否切换成功
2、运行的 PaddleSeg 项目与下载的 paddleseg 版本准确匹配
paddleseg 在 Windows 系统中只有 2.8.0 版本,因此需要下载 PaddleSeg2.8.0 的项目(因为目前更新到了 2.9.0,不要用最新的,一定要匹配)
paddleseg 在 Linux 系统中有 2.9.0 版本,因此下载 PaddleSeg2.9.0 的项目可以正常使用
当 train.py 报错出现找不到paddleseg包时,可以重新安装 paddleseg 包,并 pip install -e .
更新