問題タブ [tensorrt-python]
For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.
pip - pycuda のホイールの構築に失敗しました
コマンドで pycuda をインストールしようとしました:
pip install pycuda == 2021.1
エラーが発生しましたFailed building wheel for pycuda
私の環境は: Ubuntu: 20.04; GPU: GeForce GTX 1070 Ti; NVIDIA ドライバー: 460.80; CUDA ツールキット: 10.2.89; cuDNN: 7.6.5; パイソン: 3.6.13
pycuda
正常にインストールするにはどうすればよいですか?
nvidia - このバージョンの TensorRT は、外側の 2 次元のパディングのみをサポートします
onnx AI モデル ファイルを nvidia tensorrt ファイルに変換しようとしています。ハードウェア: NVIDIA xavier AGX。tensorrt のバージョンとハードウェア情報:
NVIDIA Jetson AGX ザビエル [16GB]
- ジェットパック 4.5.1 [L4T 32.5.1]
- NV 電力モード: MAXN - タイプ: 0
- jetson_stats.service: アクティブ
- ライブラリ:
- CUDA: 10.2.89
- cuDNN: 8.0.0.180
- テンソルRT: 7.1.3.0
- ビジョンワークス: 1.6.0.501
- OpenCV: 3.4.15-dev コンパイル済み CUDA: いいえ
- VPI: ii libnvvpi1 1.0.15 arm64 NVIDIA ビジョン プログラミング インターフェイス ライブラリ
- バルカン: 1.2.70
ただし、コードを実行すると
次のエラーが表示されます。
onnx ファイルから fp32 TensorRT エンジンをビルドしています...
ONNX ファイルの解析を開始しています
[TensorRT] 警告: onnx2trt_utils.cpp:220: ONNX モデルは INT64 の重みで生成されていますが、TensorRT は INT64 をネイティブにサポートしていません。INT32にキャストダウンしようとしています。
onnx ファイルの解析中に 1 個のエラーが見つかりました。
Onnx モデル ファイル: Conv パッド: 1, 1, 2, 2.
誰もそれを修正する方法を知っていますか?