最近为了ONNX的版本问题,又对本文所述内容进行了深入的调研。
问题背景是,待部署的ONNX model是通过特定版本的ONNX获得的,从而需要和特定版本的cuda和cudnn搭配。从这里可以查到ONNX,cuda,cudnn三者的版本搭配。
为了解决这个问题,最稳妥的方式是按照官网步骤,安装一遍cuda和cudnn. 然而这样搞非常浪费时间,而且我们知道ONNX其实只是需要特定版本的库文件。在某些比较方便的情况下,Python的包管理器conda和pip能帮我们简化安装步骤。
最近为了ONNX的版本问题,又对本文所述内容进行了深入的调研。
问题背景是,待部署的ONNX model是通过特定版本的ONNX获得的,从而需要和特定版本的cuda和cudnn搭配。从这里可以查到ONNX,cuda,cudnn三者的版本搭配。
为了解决这个问题,最稳妥的方式是按照官网步骤,安装一遍cuda和cudnn. 然而这样搞非常浪费时间,而且我们知道ONNX其实只是需要特定版本的库文件。在某些比较方便的情况下,Python的包管理器conda和pip能帮我们简化安装步骤。