Skip to content

Upgrade gpu build to CUDA 10 + cudnn 7.3 - #112

Merged
Hector Li (HectorSVC) merged 3 commits into
masterfrom
hecli/upgrade_to_cuda10
Dec 6, 2018
Merged

Hector Li (HectorSVC) merged 3 commits into
masterfrom
hecli/upgrade_to_cuda10

Conversation

@HectorSVC

Copy link
Copy Markdown
Contributor

Upgrade gpu build to CUDA 10 + cudnn 7.3
Keep the cuda 9.1 in the separate buld pipeline

@raymondxyang

Raymond Yang (raymondxyang) commented Dec 6, 2018 •

Copy link
Copy Markdown

Hi Hector could u also copy the new content in yaml to the py-packaging.yml? Thanks #Resolved

if [ $BUILD_DEVICE = "gpu" ]; then
IMAGE="ubuntu16.04-cuda9.1-cudnn7.1"
IMAGE="ubuntu16.04-$CUDA_VER"
docker build -t "onnxruntime-$IMAGE" --build-arg PYTHON_VERSION=${PYTHON_VER} -f Dockerfile.ubuntu_gpu .

@HectorSVC Hector Li (HectorSVC) Dec 6, 2018 •

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

Dockerfile.ubuntu_gpu [](start = 85, length = 21)

should switch the docker file as well, will fix it for cuda9.1 build #Resolved

inputs:
scriptPath: 'tools/ci_build/github/linux/run_dockerbuild.sh'
args: '-c Release -o ubuntu16.04 -d gpu -r $(Build.BinariesDirectory) -p $(python.version) -x "--build_wheel --use_mkldnn"'
args: '-c Release -o ubuntu16.04 -d gpu -c cuda9.1-cudnn7.1 -r $(Build.BinariesDirectory) -p $(python.version) -x "--build_wheel --use_mkldnn"'

@pranavsharma Pranav Sharma (pranavsharma) Dec 6, 2018 •

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.

did you mean to say cuda10 here? #WontFix

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

for the python package I think we are still releasing with cuda 9.1 for Azure?


In reply to: 239301044 [](ancestors = 239301044)

@pranavsharma Pranav Sharma (pranavsharma) Dec 6, 2018 •

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.

yeah, not sure if Azure upgraded to cuda 10. Vinitra Swamy (@vinitra) might know. #WontFix

@@ -0,0 +1,41 @@
jobs:
- job: Windows_CI_GPU_Dev
pool: Win-GPU-CUDA10

@pranavsharma Pranav Sharma (pranavsharma) Dec 6, 2018 •

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.

same comment as above. #WontFix

@HectorSVC
Hector Li (HectorSVC) merged commit a68f5cc into master Dec 6, 2018
@HectorSVC
Hector Li (HectorSVC) deleted the hecli/upgrade_to_cuda10 branch December 6, 2018 19:42
Ted Themistokleous (TedThemistokleous) pushed a commit to TedThemistokleous/onnxruntime that referenced this pull request Jun 2, 2025
Ted Themistokleous (TedThemistokleous) pushed a commit to TedThemistokleous/onnxruntime that referenced this pull request Jul 9, 2025
ankus-qti pushed a commit to CodeLinaro/onnxruntime that referenced this pull request Nov 25, 2025
Dmitri Smirnov (yuslepukhin) pushed a commit that referenced this pull request Mar 17, 2026
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.

3 participants