PFernandes7 opened a new issue #10426: URL: https://github.com/apache/tvm/issues/10426
After installing TVM correctly on this machine when trying to compile a model from Keras using `relay.frontend.from_keras(model, shape_dict)` defining target either as `target = tvm.target.cuda()` I get `[17:27:26] ../src/target/target_kind.cc:163: Warning: Unable to detect CUDA version, default to "-arch=sm_20" instead` or `target = tvm.target.cuda(arch="sm_20")` and then I get the same warning when calling `lib = relay.build(mod, target=target, params=params)` it again prints the same warning and exits with `Compilation error: nvcc fatal : Unsupported gpu architecture 'compute_20' ` After building it from source during the cmake or make command it seems to find the correct CUDA what might be the issue here? -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
