在GPU上使用Tensorflow出现:Cannot assign a device for operation strided_slice_2: node strided_slice_2 (defi

在GPU跑tensorflow出现了这个问题:Cannot assign a device for operation strided_slice_2: node strided_slice_2 (defined at D:\anaconda\lib\site-packages\tensorflow_core\python\framework\ops.py:1748) was explicitly assigned to /device:GPU:1 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0, /job:localhost/replica:0/task:0/device:GPU:0 ]. Make sure the device specification refers to a valid device.
[[strided_slice_2]]

这个问题出现的其中的一个原因是:tensorflow要求CUDA和CUDnn版本不匹配。
例如:
tensorflow-gpu==1.15
对于软件的要求是:
1、NVIDIA® GPU 驱动程序:CUDA 10.1 需要 418.x 或更高版本。
2、cuDNN SDK(7.6 及更高版本)
但是目前我配置的版本是CUDA10.0和CUDNN7.4.1,这将导致tensorflow-gpu无法调用CUDA10.0和CUDNN,以至出现这种报错情况。
友情提示:一定要根据自己实际运行的代码确定tensorflow-gpu的安装版本,然后在tensorflow的官方文档查看支持的CUDA10.0和CUDNN。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值