chore(docker): update base image to new cuda & torch#148
chore(docker): update base image to new cuda & torch#148ndeybach wants to merge 1 commit intoVincentqyw:mainfrom
Conversation
- from "pytorch:2.4.0-cuda12.1-cudnn9-runtime" to "pytorch/pytorch:2.9.0-cuda12.8-cudnn9-runtime"
|
@Vincentqyw , thanks for merging my other PR just now. For this one I think that from my november memory I later encountered a crash / incompatibility when testing one model (that was build as research code and never upgraded beyond 12.1 ... so there may be incompatibilities. I don't know how you would like to handle updates to newer cuda versions (which new models do need though for some aspects and acceleration / optimizations when benchmarking them on new arch)... |
|
Hi @ndeybach, to be honest, I don't have a perfect solution for version management either; it's mostly been trial and error on my end. |
Updates the base image to newer cuda and pytroch to better support newer gpu architectures.