-
Notifications
You must be signed in to change notification settings - Fork 265
feat: GPU jenkins support #476
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
Conversation
Also, please share successful build URL in description. Would be helpful for future reference |
docker/jenkins.Dockerfile
Outdated
@@ -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 |
There was a problem hiding this comment.
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
There was a problem hiding this comment.
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.
There was a problem hiding this comment.
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?
There was a problem hiding this comment.
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.
There was a problem hiding this comment.
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?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done!
Updated Dockerfile
Jenkins Job: https://ada-00.cc.gatech.edu/jenkins/job/kravicha3%20-%20Test%20Jenkins%20CI%20EVA/69/