Skip to content

Updates to CUDA and TensorRT dockerfiles for v0.5.0 - #1731

Merged
Vinitra Swamy (vinitra-zz) merged 7 commits into
microsoft:masterfrom
vinitra-zz:vinitra/docker-image-updates
Sep 13, 2019
Merged

Updates to CUDA and TensorRT dockerfiles for v0.5.0#1731
Vinitra Swamy (vinitra-zz) merged 7 commits into
microsoft:masterfrom
vinitra-zz:vinitra/docker-image-updates

Conversation

@vinitra-zz

@vinitra-zz Vinitra Swamy (vinitra-zz) commented Aug 29, 2019

Copy link
Copy Markdown

Description: This PR updates the CUDA dockerfile base image to use CUDA 10.1 and TensorRT dockerfile to use version 19.06.

Motivation and Context

@snnn Changming Sun (snnn) left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

FYI, we will revert it to 10.0.

Comment thread dockerfiles/README.md Outdated

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Update the table in dockerfile/README.md

@jywu-mysoft

Copy link
Copy Markdown
Contributor

FYI, we will revert it to 10.0.

Our thinking here was that AML has necessary drivers for CUDA 10.1,
so we can just go ahead and use 10.1 for the docker image deployed to AML.

Also, The TensorRT base image from Nvidia is using CUDA 10.1

@snnn

Copy link
Copy Markdown
Contributor

FYI, we will revert it to 10.0.

Our thinking here was that AML has necessary drivers for CUDA 10.1,
so we can just go ahead and use 10.1 for the docker image deployed to AML.

Also, The TensorRT base image from Nvidia is using CUDA 10.1

Fine. I'll ask Faith what to do. Faith Xu (@faxu)

@vinitra-zz

Copy link
Copy Markdown
Author

quick bump on this PR. @jywu-msft, Faith Xu (@faxu), Changming Sun (@snnn), manashgoswami (@manashgoswami) -- any additional input? This is ready to be merged from my side.

Comment thread dockerfiles/Dockerfile.cuda Outdated
Comment thread dockerfiles/README.md Outdated
@vinitra-zz
Vinitra Swamy (vinitra-zz) merged commit 2a21df2 into microsoft:master Sep 13, 2019
@vinitra-zz
Vinitra Swamy (vinitra-zz) deleted the vinitra/docker-image-updates branch September 13, 2019 21:16
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants