tensorflow failed call to cuInit: CUDA_ERROR_UNKNOWN

tensorflow

一个面向所有人的开源机器学习框架

项目地址：https://gitcode.com/gh_mirrors/te/tensorflow

免费下载资源

LF-DevJourney

16181人浏览 · 2018-07-01 12:41:11

LF-DevJourney · 2018-07-01 12:41:11 发布

Tensorflow 之前使用GPU还是好好的，但是今天突然发现不了GPU设备了，同事在执行

tf.Session() 的时候抛出下面的错误：

tensorflow/stream_executor/cuda/cuda_driver.cc:406 failed call to cuInit: CUDA_ERROR_UNKNOWN

具体解决方案参考我在 StackOverflow 的 Answer: failed call to cuInit: CUDA_ERROR_UNKNOWN

附：
Tensorflow-GPU 详细安装及开发环境搭建

GitHub 加速计划 / te / tensorflow

下载

一个面向所有人的开源机器学习框架

最近提交(Master分支：1 个月前 )

4f64a3d5 Instead, check for this case in `ResolveUsers` and `ResolveOperand`, by querying whether the `fused_expression_root` is part of the `HloFusionAdaptor`. This prevents us from stepping into nested fusions. PiperOrigin-RevId: 724311958 3 个月前

aa7e952e Fix a bug in handling negative strides, and add a test case that exposes it. We can have negative strides that are not just -1, e.g. with a combining reshape. PiperOrigin-RevId: 724293790 3 个月前