search

tlc-pack / tlcpack

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

Building images for CUDA 11 #70

Closed shyam196 closed 1 year ago

shyam196 commented 2 years ago

Thanks for building these packages -- it is a huge time saver!

Is there any interest in building packages for CUDA 11 series? I'm happy to look into this myself and upstream if you are keen :-)

tqchen commented 2 years ago

Thanks @shyam196 , yes, indeed we would love to build cuda 11 series. cc @areusch @leandron (we might want to also add the https://github.com/tlc-pack/tlcpack/tree/main/docker to nightly so it is easier to obtain the tlcpack building images)

shyam196 commented 2 years ago

I'll take a look for you this week hopefully.

To clarify, is the rough procedure:

  1. Add a new Dockerfile (extending pytorch/manylinux-cu111 for example)
  2. Add CUDA 11 as an option in the relevant scripts / action matrix?
DoTriN7 commented 2 years ago

Hi @shyam196 @tqchen, just to know, is there an ETA of the packages built for CUDA 11 series?

areusch commented 2 years ago

@DoTriN7 we'll see if we can look at this in the next month or so

areusch commented 2 years ago

cc @hpanda-naut