Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Update GPU install Ubuntu 16.04 #1421

Merged
merged 3 commits into from Jan 25, 2020
Merged

Update GPU install Ubuntu 16.04 #1421

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
b8c62c0
Clarify that CUDA Ubuntu 16.04 Install is for 10.1
sarthfrey Jan 15, 2020
34c5f7d
Add libnvinfer-plugin for TensorRT
lamberta Jan 22, 2020
1778ce3
Update to libnvinfer6
lamberta Jan 22, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
10 changes: 6 additions & 4 deletions site/en/install/gpu.md
View file
Open in desktop
Expand Up @@ -119,7 +119,7 @@ complicates installation of the NVIDIA driver and is beyond the scope of these i
</pre>


#### Ubuntu 16.04 (CUDA 10)
#### Ubuntu 16.04 (CUDA 10.1)

<pre class="prettyprint lang-bsh">
# Add NVIDIA package repositories
Expand All @@ -141,14 +141,16 @@ complicates installation of the NVIDIA driver and is beyond the scope of these i

# Install development and runtime libraries (~4GB)
<code class="devsite-terminal">sudo apt-get install --no-install-recommends \
cuda-10-0 \
cuda-10-1 \
libcudnn7=7.6.4.38-1+cuda10.1 \
libcudnn7-dev=7.6.4.38-1+cuda10.1
</code>

# Install TensorRT. Requires that libcudnn7 is installed above.
<code class="devsite-terminal">sudo apt-get install -y --no-install-recommends libnvinfer5=6.0.1-1+cuda10.1 \
libnvinfer-dev=6.0.1-1+cuda10.1
<code class="devsite-terminal">sudo apt-get install -y --no-install-recommends \
libnvinfer6=6.0.1-1+cuda10.1 \
libnvinfer-dev=6.0.1-1+cuda10.1 \
libnvinfer-plugin6=6.0.1-1+cuda10.1
</code>
</pre>

Expand Down