-
Notifications
You must be signed in to change notification settings - Fork 76
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
OSError: CUDA_HOME environment variable not set when python setup.py in Dockerfile #95
Comments
Hi, looks like to met that you would need to use the |
Hi @ClementPinard Thank you for your advice
|
See this related issue : #90 GPU is not available during docker build so you need to figure out your compute capbilities beforehand and set the |
Hi @ClementPinard Thank you for your solution |
If you don't know what the gpu cuda capabilties of your machine will be, your best bet is to compile for as much architectures as possible, or wait for the docker to be launched to compile the library. Compiled code cannot be generic |
My Dockerfile
Then raise an Error:
OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.
The full error logs:
The text was updated successfully, but these errors were encountered: