小白踩坑,記錄一下。tensorflow2.0.0-beta1版本蒸苇,cuda版本10.0.130,cudnn版本7.6.3.30吮旅。
在學(xué)習(xí)tensorflow2.0時(shí)溪烤,學(xué)習(xí)鏈接:簡(jiǎn)單粗暴TensorFlow2.0:使用 Keras 實(shí)現(xiàn)卷積神經(jīng)網(wǎng)絡(luò)
卷積神經(jīng)網(wǎng)絡(luò)(CNN)這部分的例子
替換以后運(yùn)行報(bào)錯(cuò):
UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. [Op:Conv2D]
解決辦法:
添加下面兩行
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '/gpu:0'