TensorRT python包与python 3.6不兼容

Question

如果您安装了TensorRT 4，则从其tar文件安装CUDA using the run file是唯一可用的选项。但是，tar文件只包含python 2.7和3.5的python TensorRT轮文件。没有提供python 3.6 wheel文件。

我无法强制在我的python 3.6系统上安装python 3.5 wheel文件：

$ pip install tensorrt-4.0.0.3-cp35-cp35m-linux_x86_64.whl 
tensorrt-4.0.0.3-cp35-cp35m-linux_x86_64.whl is not a supported wheel on this platform.

另外，我不能创建一个新的python 3.5 virtualenv只是为了使用TensorRT。

有没有办法通过手动复制文件在我的python 3.6平台上手动安装这个软件包？

Answer 1

是的，您可以将此文件重命名为tensorrt-4.0.0.3-cp36-cp36m-linux_x86_64.whl，然后重试。

有效。

Answer 2

请使用python版本3.5.5。我在python版本3.6中遇到了同样的问题。最新版本的tensorflow适用于python 3.5。如果你使用anaconda使用conda shell来激活早期的python版本。您可以设置环境以使用早期版本的python。