search

tlc-pack / tlcpack

https://tlcpack.ai/
Apache License 2.0
23 stars 30 forks source link

[CUDA] Deprecate 10.0/10.1 and add 11.0/11.1/11.3 #121

Closed comaniac closed 2 years ago

comaniac commented 2 years ago

Closing #120

@areusch please let me know if we need to manually build and push the docker images for CUDA 11 series. Also, I'm not sure if we need to change the version set in the TLCPack website manually?

areusch commented 2 years ago

@comaniac i think we probably need to adjust https://github.com/tlc-pack/tlc-pack.github.io/blob/main/js/quick-start-widget.js#L124

areusch commented 2 years ago

i think we merge this and trigger the nightly build and then we should see the CI go green here. doing that now...

areusch commented 2 years ago

mm, nope we need to build the package-11x images. doing that now

comaniac commented 2 years ago

I found that the base images of CUDA 11 from PyTorch do not change the default CUDA path. Please let me know if that affects the tlcpack build. If so we may need a patch to the docker file:

RUN rm /usr/local/cuda; ln -s /usr/local/cuda-11.0 /usr/local/cuda
areusch commented 2 years ago

hmm, seems like we got some errors: https://github.com/tlc-pack/tlcpack/runs/7350404897?check_suite_focus=true

comaniac commented 2 years ago

Hmm looks like there are some dependency issues with Conda build in torch CUDA 11.0 image. I'll investigate it. Meanwhile, feel free to revert this PR if it blocks anything on going.