在运用tf_ops算子时出现InvalidArgumentError (see above for traceback): No OpKernel was registered to support

在运用tf_ops算子时出现InvalidArgumentError (see above for traceback): No OpKernel was registered to support Op ‘FarthestPointSample’ with these attrs. Registered devices: [CPU], Registered kernels:device=‘GPU’

原因:TensorFlow无法调用你的GPU,可能时的显卡驱动除了问题。我是将电脑重启之后,显卡就能被重新调用了。具体可详见:https://github.com/charlesq34/pointnet2/issues/33

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
TensorFlow是一个非常强大的深度学习框架,支持的操作非常多,但有某些特殊的操作没有被包含在标准库中,需要使用者自己手动编译源代码。如果想在Winodws系统下编译tf_ops,需要遵循以下步骤: 1. 安装Visual Studio Visual Studio是一个从微软公司发布的IDE,用来开发、调试和部署Windows程序,其中包括了C++编译器。由于TensorFlow是用C++实现的,所以需要安装Visual Studio。在Windows下,可以前往微软官网下载最新的Visual Studio Community版本,按照安装向导进行安装即可。 2. 安装CUDA CUDA是NVIDIA发布的一款并行计算平台和编程模型,是TensorFlow的必要条件。需要选择与显卡硬件相适应的CUDA版本,并安装后加入环境变量PATH。 3. 安装cuDNN cuDNN是NVIDIA针对深度神经网络运算的加速库,可以提高TensorFlow的计算效率。需要到NVIDIA官网注册后下载与CUDA版本相匹配的cuDNN,并添加到环境变量PATH。 4. 编译源代码 首先需要下载TensorFlow源代码,然后在命令行窗口进入到该源代码目录中。执行以下命令: ```bash cd tensorflow # 进入到源代码目录 bazel build -c opt --config=cuda //tensorflow/core/user_ops:path/to/user_ops ``` 其中`path/to/user_ops`需要被替换成自己存放user ops源代码的目录。此命令会自动编译user ops的源代码并生成.so文件,可以用于TensorFlow的其他程序中使用。 总之,Windwos下编译tf_ops需要安装Visual Studio、CUDA和cuDNN,并按照TensorFlow官方文档中的步骤进行编译即可。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值