search

Jittor / jittor

Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.
https://cg.cs.tsinghua.edu.cn/jittor/
Apache License 2.0
3.08k stars 311 forks source link

Runtime error #241

Open Annada666 opened 3 years ago

Annada666 commented 3 years ago

When I import jittor in the program, something happened. [i 0705 19:52:00.941816 12 compiler.py:869] Jittor(1.2.3.57) src: /home/yanjing/anaconda3/lib/python3.8/site-packages/jittor [i 0705 19:52:00.946074 12 compiler.py:870] g++ at /usr/bin/g++(5.4.0) [i 0705 19:52:00.946285 12 compiler.py:871] cache_path: /home/yanjing/.cache/jittor/default/g++ [i 0705 19:52:00.951435 12 init.py:286] Found /usr/local/cuda/bin/nvcc(10.0.130) at /usr/local/cuda/bin/nvcc. [i 0705 19:52:01.020187 12 init.py:286] Found gdb(16.04.1) at /usr/bin/gdb. [i 0705 19:52:01.024134 12 init.py:286] Found addr2line(2.26.1) at /usr/bin/addr2line. [i 0705 19:52:01.040835 12 compiler.py:958] py_include: -I/home/yanjing/anaconda3/include/python3.8 -I/home/yanjing/anaconda3/include/python3.8 [i 0705 19:52:01.056686 12 compiler.py:960] extension_suffix: .cpython-38-x86_64-linux-gnu.so [i 0705 19:52:01.240809 12 compiler.py:1093] OS type:ubuntu OS key:ubuntu [i 0705 19:52:01.242638 12 init.py:178] Total mem: 251.56GB, using 16 procs for compiling. [i 0705 19:52:01.374744 12 jit_compiler.cc:22] Load cc_path: /usr/bin/g++ [i 0705 19:52:01.627651 12 init.cc:55] Found cuda archs: [75,] [i 0705 19:52:01.710848 12 compile_extern.py:444] mpicc not found, distribution disabled. [i 0705 19:52:01.779519 12 compile_extern.py:20] found /usr/local/cuda/include/cublas.h [i 0705 19:52:01.790217 12 compile_extern.py:20] found /usr/local/cuda/lib64/libcublas.so terminate called after throwing an instance of 'std::runtime_error' what(): [f 0705 19:52:02.076052 12 helper_cuda.h:126] CUDA error at /home/yanjing/anaconda3/lib/python3.8/site-packages/jittor/extern/cuda/cublas/src/cublas_warper.cc:21 code=1( CUBLAS_STATUS_NOT_INITIALIZED ) cublasCreate(&cublas_handle) Aborted (core dumped)

Jittor commented 3 years ago

it seems your cuda installation may have problems, try python3.8 -m jittor_utils.install_cuda and running again? jittor will try to fix cuda installation.

Annada666 commented 3 years ago

Thanks. It works.