- 博客(3)
- 收藏
- 关注
原创 jetson安装pycuda报错的解决方案
在使用pip安装pycuda的时候可能会遇见好几种报错 1.找不到cuda.h头文件 首先是最常见的 src/cpp/cuda.hpp:14:10: fatal error: cuda.h: No such file or directory #include <cuda.h> ^~~~~~~~ compilation terminated. error: command 'aarch64-linux-gnu-gcc' failed with exit
2021-04-08 16:54:19
1700
2
原创 关于tensorflow的报错NodeDef mentions attr ‘xxx‘ not in Op的解决方案和产生原因
在尝试使用下载的pb文件(frozen graph)时遇上了这个问题,报错信息如下。 Traceback (most recent call last): File "/home/shuaige/anaconda3/envs/tf15/lib/python3.7/site-packages/tensorflow_core/ python/framework/importer.py", line 501, in _import_graph_def_internal graph._c_graph,
2021-03-11 11:43:08
2735
3
原创 关于Could not load dynamic library ‘libnvinfer.so.6‘和 ‘libcudart.so.10.0‘的解决
前言 首先,现在可以直接在tensorflow的工作流中调用tensorRT的代码了,统称为TF-TRT,不需要再单独使用TensorRT进行优化,但运行需要本地安装了TensorRT 如果想在运用TensorFlow中自带的TensorRT,例如以下代码 from tensorflow.python.compiler.tensorrt import trt_convert as trt input_saved_model_dir = "saved_model_dir" converter = trt
2020-12-16 11:16:23
1962
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人