docker安装tenserRT
时间: 2023-10-09 18:07:15 浏览: 58
要安装TensorRT,需要先安装Docker和NVIDIA Docker。首先,您需要使用以下命令安装Docker:
1. 更新软件包列表:
```
sudo apt-get update
```
2. 安装Docker CE:
```
sudo apt-get install docker-ce
```
3. 验证Docker安装是否成功:
```
docker --version
```
接下来,您需要安装NVIDIA Docker,以便在Docker中使用GPU:
1. 添加NVIDIA Docker存储库的GPG密钥:
```
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
```
2. 添加NVIDIA Docker存储库:
```
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
```
3. 更新软件包列表:
```
sudo apt-get update
```
4. 安装NVIDIA Docker:
```
sudo apt-get install -y nvidia-docker2
```
安装完成后,您可以使用以下命令验证NVIDIA Docker是否安装成功:
```
docker run --gpus all nvidia/cuda:11.0-base nvidia-smi
```
现在,您可以安装TensorRT并在Docker容器中使用它:
1. 使用以下命令启动TensorRT Docker容器:
```
docker run --gpus all -it --rm -v local_dir:container_dir nvcr.io/nvidia/tensorrt:xx.xx-py3
```
在上述命令中,将"local_dir"替换为本地目录的路径,将"container_dir"替换为容器中的目录路径,并将"xx.xx"替换为您要使用的TensorRT版本号。
这样,您就可以在Docker容器中安装和使用TensorRT了。请注意,您可能还需要根据您的具体需求配置和运行TensorRT。<span class="em">1</span><span class="em">2</span><span class="em">3</span><span class="em">4</span>
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)