We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
You can set as such
int main(int argc, char **argv) { cudaError_t cudaStatus = cudaSetDevice(GPU_INDEX); if (cudaStatus != cudaSuccess) { fprintf(stderr, "cudaSetDevice failed! Do you have a CUDA-capable GPU installed? Index: %d\n", GPU_INDEX); return 1; } ... }
The text was updated successfully, but these errors were encountered:
Actually not certain if this works or not.
This worked for sure though
export CUDA_VISIBLE_DEVICES=<index of gpu>
Sorry, something went wrong.
No branches or pull requests
You can set as such
The text was updated successfully, but these errors were encountered: