search

yuh-zha / AlignScore

ACL2023 - AlignScore, a metric for factual consistency evaluation.
https://arxiv.org/pdf/2305.16739.pdf
MIT License
110 stars 18 forks source link

Unable to load model in GPU during docker build process #19

Open Ryan-ZL-Lin opened 2 months ago

Ryan-ZL-Lin commented 2 months ago

Hi. I'd like to use GPU to load model to build Docker image. Here is the parameter I changed in Dockerfile form this repo image

and here is my GPU spec and driver by running nvidia-smi image

However, during the Docker build process, I got this error saying 

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

Does anyone know how to better address this issue?