2021年4月1日
bazel编译tvm和cuda
摘要： 在docker中编译tvm和cuda，工程是tensorflow-serving所以要使用bazel编译器，在docker中可以看到显卡驱动和cuda版本，但是编译的时候总是找不到头文件 后来做了个软连接，将cuda目录映射到库目录下就解决了 ln -s /usr/local/cuda-11.0/i 阅读全文
posted @ 2021-04-01 16:36 Wangtn 阅读(623) 评论(0) 推荐(0) 编辑