前几天新建恒源云服务器实例的时候,发现cuda版本已经支持到11.8,表明已经满足编译安装katago1.13版本的条件了。
于是进行了升级,顺利完成。
首先,要确定实例环境。现在恒源云还是有些混乱,部分实例的显卡驱动版本还是510.30.02,这种实例是无法安装高版本cuda的。
下载tensorrt版本后,进行安装
dpkg -i nv-tensorrt-local-repo-ubuntu2004-8.6.1-cuda-11.8_1.0-1_amd64.deb
cp /var/nv-tensorrt-local-repo-ubuntu2004-8.6.1-cuda-11.8/nv-tensorrt-local-D7BB1B18-keyring.gpg /usr/share/keyrings/
apt update
apt install tensorrt
dpkg-query -W tensorrt
编译katago
git clone https://github.com/lightvector/KataGo.git
cd KataGo/cpp
cmake . -DUSE_BACKEND=TENSORRT
make
在AutoDL服务器中,目前已经有官方Tensorrt8.5的镜像,这样就免去了tensorrt的安装。
网友评论