CUDA error at ~/.local/lib/python3.8/site-packages/jittor/src/ops/array_op.cc:32 code=222( cudaErrorUnsupportedPtxVersion ) cudaStreamCreateWithFlags(&stream, cudaStreamNonBlocking)

[jiapei100]

Hi:

My env:

• OS: Ubuntu 20.04
• Cuda: 11.3
• cudnn 8.2.0
• gcc/g++: 10.2

Failed to run test code as:

  DeepFaceEditing-Jittor git:(main) python test_model.py --geo ./results/sketch_gen.png --appear ./images/appearance.png --output ./results/sketch_result.png --geo_type sketch
[i 0512 11:37:12.017235 64 compiler.py:851] Jittor(1.2.2.71) src: ~/.local/lib/python3.8/site-packages/jittor
[i 0512 11:37:12.017286 64 compiler.py:852] g++ at /usr/bin/g++
[i 0512 11:37:12.017319 64 compiler.py:853] cache_path: ~/.cache/jittor/default/g++
[i 0512 11:37:12.019891 64 __init__.py:257] Found /usr/local/cuda/bin/nvcc(11.3.58) at /usr/local/cuda/bin/nvcc.
[i 0512 11:37:12.070950 64 __init__.py:257] Found gdb(9.2) at /usr/bin/gdb.
[i 0512 11:37:12.074234 64 __init__.py:257] Found addr2line(2.34) at /usr/bin/addr2line.
[i 0512 11:37:12.108394 64 compiler.py:893] pybind_include: -I/usr/include/python3.8 -I~/.local/lib/python3.8/site-packages/pybind11/include
[i 0512 11:37:12.116724 64 compiler.py:895] extension_suffix: .cpython-38-x86_64-linux-gnu.so
[i 0512 11:37:12.207914 64 __init__.py:169] Total mem: 31.27GB, using 10 procs for compiling.
terminate called after throwing an instance of 'std::runtime_error'
  what():  [f 0512 11:37:12.404950 64 helper_cuda.h:126] CUDA error at ~/.local/lib/python3.8/site-packages/jittor/src/ops/array_op.cc:32  code=222( cudaErrorUnsupportedPtxVersion ) cudaStreamCreateWithFlags(&stream, cudaStreamNonBlocking)

Any suggestions?

[tiwarikaran]

I tried it on collab and it's working alright, although I have some doubts