-
Notifications
You must be signed in to change notification settings - Fork 3.7k
[CI] Update GPU image for PyTorch 1.11 #10849
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
ecd4ba2 to
02c9df9
Compare
| from matplotlib import pyplot as plt | ||
| from keras.applications.resnet50 import preprocess_input | ||
| from tensorflow.keras.applications.resnet50 import preprocess_input | ||
|
|
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.
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.
also not sure, the mapping of image version -> range of commits is very difficult to track down (but should be fixed by #10646)
| from matplotlib import pyplot as plt | ||
| from keras.applications.resnet50 import preprocess_input | ||
| from tensorflow.keras.applications.resnet50 import preprocess_input | ||
|
|
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.
also not sure, the mapping of image version -> range of commits is very difficult to track down (but should be fixed by #10646)
* Update Jenkinsfile to point to the new GPU image * fix onnx test * fixed keras tutorial
* Update Jenkinsfile to point to the new GPU image * fix onnx test * fixed keras tutorial
To support the new PT fallback feature, we need to update PT to 1.11 (for dlpack bump). See more discussion in #10758
Validated on https://ci.tlcpack.ai/blue/organizations/jenkins/tvm/detail/ci-docker-staging/244/pipeline
@t-vi @driazati @areusch @leandron @Mousius