恒源云下安装katago所需TensorRT环境

外出两周,回家后本想进入恒源云继续整理中断的工作,结果发现原先使用的实例不见了。

原来,恒源云的实例保留时间是十天,而不是AutoDL的30天。

只好重新创建实例,并安装TensorRT环境。之前在公众号中也写过配置文件,但只是简要说明,安装的时候耽误了些许时间,本文还是详细介绍一下,给自己也给需要朋友以供日后参考。

毕竟GPU云服务器,时间就是金钱。

创建新实例,使用官方镜像:TensorFlow / 2.5.0 / 11.2 / 3.8

按照Nvidia官方安装说明,安装TensorRT:

python3 -m pipinstall–upgrade setuptools pip

python3 -m pipinstallnvidia-pyindex

python3 -m pipinstall–upgrade nvidia-tensorrt

安装完毕后,编译katago需要的库文件被安装在:

/usr/local/lib/python3.8/dist-packages/tensorrt

可以设置系统参数,将默认参数指向此路径,最为省劲的方法,还是直接将此路径下所有*.so.8文件,还有libnvinfer_builder_resource.so.8.4.1拷贝至/usr/local/lib目录下。

同时拷贝到/usr/local/lib目录下的还有libzip.so.5。

上传katago官网已经编译好的katago-v1.10.0-trt8.2-cuda11.1-linux-x64版本文件,即可运行katago。

Leave a Reply

您的电子邮箱地址不会被公开。 必填项已用 * 标注

You may use these HTML tags and attributes:

<a href="" title=""> <abbr title=""> <acronym title=""> <b> <blockquote cite=""> <cite> <code> <del datetime=""> <em> <i> <q cite=""> <s> <strike> <strong>