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

源代码 · 2024-9-7 21:18:10

外出两周，回家后本想进入恒源云继承整理制止的工作，结果发现原先利用的实例不见了。
原来，恒源云的实例生存时间是十天，而不是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。

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

所属分类: 问答交流

新帖推荐: 30日

推荐作品