(base) ryanyao@umihuang-HP-EliteDesk-800-G1-TWR:~/docker_mount/tvm_ex$ python tensor_expr_get_started.py
<class ‘tvm.tensor.Tensor’>
terminate called after throwing an instance of ‘dmlc::Error’
what(): [17:27:23] /home/ryanyao/tvm/src/codegen/opt/build_cuda_on.cc:118: Check failed: compile_res == NVRTC_SUCCESS (5 vs. 0) : nvrtc: error: invalid value for --gpu-architecture (-arch)
Aborted (core dumped)
Can someone give me hint how to solve it?
Thank you.
Remove tvm and get new tvm source to build it. Because my root cause is that there are some problme (assing target) during building the tvm previously.
The nvcc -V reports a version as 9.0.176, but the nvidia-smi reports a version 10.1. My tensorflow version is 1.12.0. The directory of /usr/local/ includes the following files:
bin cuda cuda-9.0 etc games include lib man sbin share src
My GPU is GTX 1660 Ti which is not supported by nvidia-384, nvidia-driver-390 that uses cuda 9.x as their runtime. I indeed tried installing the following drivers nvidia-driver-418, nvidia-driver-430, nvidia-driver-435 which all uses cuda 10.x, but the same issue emerged.