I have added a Dockerfile + relevant options to enable building using a CUDA 11.1 image. I have also updated the Github workflows.
It turns out that the upstream CUDA 11.1 image contains both a 10.2 and 11.1 installation, which caused issues as the default CUDA install was symlinked to 10.2. Once this is fixed, compilation went smoothly.
I don't believe that the Github action will run just yet: I think a maintainer needs to build the image and tag it appropriately?
Finally - I have not been able to build the conda package successfully (but I cannot do this for the existing packages either). The shared libraries do not appear to get bundled into the wheel, causing a dynamic linking issue when you try to import tvm in Python
I have added a Dockerfile + relevant options to enable building using a CUDA 11.1 image. I have also updated the Github workflows.
It turns out that the upstream CUDA 11.1 image contains both a 10.2 and 11.1 installation, which caused issues as the default CUDA install was symlinked to 10.2. Once this is fixed, compilation went smoothly.
I don't believe that the Github action will run just yet: I think a maintainer needs to build the image and tag it appropriately?
Finally - I have not been able to build the conda package successfully (but I cannot do this for the existing packages either). The shared libraries do not appear to get bundled into the wheel, causing a dynamic linking issue when you try to import tvm in Python