ubuntu tensorrt安装 cuda11.6
时间: 2023-09-26 15:14:59 浏览: 125
要在Ubuntu上安装TensorRT和CUDA 11.6,你可以按照以下步骤进行操作:
1. 首先,你需要安装nv-tensorrt-repo的deb文件。使用以下命令将文件安装到系统中:
```
sudo dpkg -i nv-tensorrt-repo-${os}-${tag}_1-1_amd64.deb
```
2. 安装TensorRT的密钥。使用以下命令将密钥添加到系统中:
```
sudo apt-key add /var/nv-tensorrt-repo-${os}-${tag}/*.pub
```
3. 更新apt源并安装TensorRT:
```
sudo apt-get update
sudo apt-get install tensorrt
```
4. 如果你只需要TensorRT而不使用TensorFlow,你可以安装numpy等所需的Python库:
```
python3 -m pip install numpy
```
5. 安装TensorRT的开发库:
```
sudo apt-get install python3-libnvinfer-dev
sudo apt-get install onnx-graphsurgeon
sudo apt-get install tensorrt-libs
sudo apt-get install tensorrt-dev
```
6. 如果你想保持旧版CUDA与TensorRT关联,可以使用以下命令:
```
sudo apt-mark hold tensorrt-dev
```
如果以后想取消关联,可以运行以下命令:
```
sudo apt-mark unhold tensorrt-dev
```
以上就是在Ubuntu上安装TensorRT和CUDA 11.6的步骤。希望对你有帮助!
阅读全文