Docker issues with PyPI installation

[sedrick-keh-tri]

When I run pip install megablocks, I seem to be getting this error:
RuntimeError: ('Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx', "This image doesn't seem to support current EC2 instance type, please check release notes for supported EC2 instance type")

However, it works fine when I build from source (i.e. pip install git+https://github.com/stanford-futuredata/megablocks.git@main)

[Trevor Gale]

Hi! Apologies for the delay!

In the case where you see the error do you have a GPU in the machine? Is the issue happening when you try to build an image for distribution?

[sedrick-keh-tri]

Hi, yeah, there's a GPU in the machine. The issue happens during building the image, yes (specifically in the step where the requirements are being installed)

[Trevor Gale]

I think this is the same as the issue described here? i.e., the GPU won't be available during docker build .... I think the solution is described here. Could you give that a try?