tensorflow-gpu测试报错：tensorflow.python.framework.errors_impl.UnknownError: 2 root error(s) found

 

目录

 

一、报错内容：

二、解决：

---

一、报错内容：

tensorflow.python.framework.errors_impl.UnknownError: 2 root error(s) found.
  (0) Unknown: 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.
	 [[{{node conv2d_24/convolution}}]]
  (1) Unknown: 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.
	 [[{{node conv2d_24/convolution}}]]
	 [[conv2d_46/truediv/_417]]
0 successful operations.
0 derived errors ignored.

二、解决：

因为是单GPU训练，所以需要加上这一行代码

os.environ["CUDA_VISIBLE_DEVICES"] = "1"

os.environ["CUDA_VISIBLE_DEVICES"] = "1"

三、再次运行：

测试成功