![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
文章平均质量分 95
白羊007
这个作者很懒,什么都没留下…
展开
-
ubuntu下TensorRT 7/8环境配置及安装
为了部署YOLO V5的模型,部署了两次TensorRT,每次都踩了不少坑,决定把完整的过程及踩过的坑记录下来。一些需要提前强调的点:所选择的TensorRT版本一定要与自己系统中的CUDA相对应(目前最新的是支持到了11.2,但不建议选择11.2,之前尝试过,无法配置成功;因此我选择了安装CUDA 11.1) 仅支持以deb方式安装的CUDA(以runfile方式安装的CUDA会遗漏一些组件,会出现如下的报错)。还看到一种说法是从某个cuda版本开始后,.run安装会把libcublas.so这原创 2021-09-24 14:18:31 · 1552 阅读 · 1 评论 -
ubuntu18.04 安装CUDA11.0+cuDNN8.0.5+TensorFlow2.4.0
前几天在系里申请了个云计算资源,准备用来跑一个论文里的模型,系统是全新的ubuntu18.04.1,需要自己配一下环境。1.安装/更新GPU驱动输入命令查看显卡信息,以及是否已经安装驱动。nvidia-smi可以看到有两块Tesla V100,驱动也都装好了,版本是418的,有点老了,注意有一个“CUDA Version: 10.1”,说明这个驱动只能安装CUDA10.1及以下的版本。所以要先卸载原来的驱动,再安装更高级的版本(如果原来没有驱动,直接安装就可以了);卸载及安装.原创 2021-01-27 14:49:54 · 9017 阅读 · 13 评论