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.

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

feat: GPU jenkins support #476

Merged
merged 2 commits into from
Nov 7, 2022
Merged

feat: GPU jenkins support #476

merged 2 commits into from
Nov 7, 2022

Conversation

gaurav274
Copy link
Member

@gaurav274 gaurav274 commented Nov 6, 2022

@kaushikravichandran
Copy link
Collaborator

Also, please share successful build URL in description. Would be helpful for future reference

@@ -0,0 +1,49 @@
# https://www.jenkins.io/doc/book/installing/docker/ - Follow these steps for starting jenkins.
FROM nvidia/cuda:11.4.2-cudnn8-devel-ubuntu20.04
Copy link
Collaborator

@kaushikravichandran kaushikravichandran Nov 6, 2022

Choose a reason for hiding this comment

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

ada-00 is using 18.04 right? We can use the 18.04 image if so

Copy link
Member Author

Choose a reason for hiding this comment

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

It is the docker image so it does not matter.

Copy link
Contributor

Choose a reason for hiding this comment

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

Any reason why you're using cuda 11.4?

Copy link
Member Author

Choose a reason for hiding this comment

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

Nope. Didn't try others. The current GPU on ada-00 works with 11.4 so I went with it.

Copy link
Contributor

Choose a reason for hiding this comment

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

The latest torchvision works best with cuda 11.6, I'm not entirely sure it works with 11.4. Can we try using 11.6 here for better compatibility?

Copy link
Member Author

Choose a reason for hiding this comment

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

Done!

Updated Dockerfile
@gaurav274 gaurav274 merged commit 821d59d into master Nov 7, 2022
@gaurav274 gaurav274 deleted the jenkins-gpu-support branch November 7, 2022 16:17
@jarulraj jarulraj mentioned this pull request Nov 21, 2022
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