求助 - tensorflow吧

level 1

whov0112 楼主

每次只要开始运行显存就直接满了，查到的解决方法是设置tf按需占用可是代码放上去后没有效果。
显卡为rtx3080，一开始装的cuda10.0没法用，后来查到30系需要装cuda11.1，cudnn8.0.5，tf-nightly-gpu2.5，torch-nightly1.8。
tf.test.is_built_with_cuda返回为true

2021年01月22日 08点01分 1