啊,适配tensorflow和3060Ti,我暴走了!
我暴走了!
感觉太糟糕了!
搞了一晚上!
早有了conda环境!
安装faceswap!
啊哈哈,自动创建了faceswap环境!不错!
什么?tensorflow是2.3?!
但是2.3不支持RTX 3060ti啊!!!
啊!不能忍!必须升级tensorflow!
为了用上RTX 3060ti,我要安装了 tensorflow-gpu 2.4!
啊,conda不支持啊,那就pip安装吧!
安装成功了!
哎呀,有点兼容问题,那就凑活用吧!
诶呀!
为了用上3060ti我还要安装cuda11.0
conda支持cuda11.0,好!
诶呀,还需要cudnn8!
但是conda不支持!
这不是配套的么?!
气死了!
好吧,我自己复制cudnn64_8.dll过去!
啊!tensorflow就是找不到这个dll!
我卸载了conda的cuda!
官网下载cuda11.0!
安装cuda11.0!
然后再把cudnn装上!
什么下载要注册?!
注册就注册!
注册完成了!
下载cudnn完成了!!
啊,是cudnn8.0.5!!
差不啊,tensorflow2.4是和cudnn8.0.2编译的!
就用这个版本吧!
安装完成了!
好!
太好了!
tensorflow已经可以识别!
可以运行啦!
功夫不负有心人!
什么不过五秒就退出了!
什么,8G内存不够用!
啊,又是兼容性问题!
cpu版本是可以跑的,而且只用了3G内存!
啊啊啊啊!
下载cudnn8.0.2!
好!
又下载完成了!
安装完成了!
什么?还是显存不够??!!
删了tensorflow-gpu 2.4!
重新安装faceswap!
重置tensorflow!
啊,怎么自动装了cuda10.1!
还有cudnn7?!
什么程序跑起来了?!
tensorflow2.4 + cuda10.1 + 3060ti!
不对啊?!
申请了内存,8G都用了!
但是cuda速度还是0!
看来确实不行!
天亮啦?!
天亮了!!
不指望,gpu啦!
cpu能跑起来就行!
哎!
太难了!
在faceswap里屏蔽掉gpu:0!
不行了!
运行了一秒就退出了!
再次运行!
好了!
正在运行。。。
我累了。。。
感觉太糟糕了。。。
联想到几天前,做pytorch夸张的内存占用率!
要么cuda,要么cudnn, 要么driver,要么显卡,要么pytorch,要么我自己肯定有问题!!!!!!
====
两天后!
升级驱动程序到461.09
发现把cuda11.0.1 升级到cuda11.0.3(cuda 11.0 udpate 1)
然后就可以运行了!
所以
windows10 + rtx 3060ti + driver461.09 + cuda11.0.3 + cudnn8.0.2 + tensorflow2.4 + tensorflow option: allow_growth
解决问题!