记录一下在Windows环境下配tensorflow1.13并运行某图像处理算法遇到的各种bug和问题和解决方法
cudnn7.6.5(本来想用7.4,因为是经过验证的可适配,但是一下子没有合适的channel提供这个版本,也不想自己手动官网下载)遇到的问题包括但不限于图像的维度问题、灰度问题、tensorflow安装的过程、cuda&cudnn选择、显卡爆满无法计算,等。安装tensorflow并且去跑机器学习算法最关键的是各个版本的cuda和cudnn要兼容,否则就会寄掉。机缘巧合下要去配Windows的tensorflow环境,还是很害怕这种事情的。算法的filter只能接收1维的图像,不能接收3维的。
原创
2024-04-17 19:30:28 ·
3595 阅读 ·
5 评论