按照Github教程使用pip安装jtorch显示安装成功
运行 快速入门部分的代码时出现断言失败。此时我的原始环境torch也被搞坏,无法正常运行代码。
(xin) xinjialong@flashxin:~$ python test.py
[i 0522 20:50:45.406909 64 compiler.py:955] Jittor(1.3.7.16) src: /home/xinjialong/anaconda3/envs/xin/lib/python3.7/site-packages/jittor
[i 0522 20:50:45.411800 64 compiler.py:956] g++ at /usr/bin/g++(11.3.0)
[i 0522 20:50:45.411932 64 compiler.py:957] cache_path: /home/xinjialong/.cache/jittor/jt1.3.7/g++11.3.0/py3.7.4/Linux-5.19.0-4x3e/IntelRXeonRW-2xe2/default
[i 0522 20:50:45.433584 64 install_cuda.py:93] cuda_driver_version: [12, 0]
[i 0522 20:50:45.439515 64 __init__.py:411] Found /home/xinjialong/.cache/jittor/jtcuda/cuda11.2_cudnn8_linux/bin/nvcc(11.2.152) at /home/xinjialong/.cache/jittor/jtcuda/cuda11.2_cudnn8_linux/bin/nvcc.
[i 0522 20:50:45.516650 64 __init__.py:411] Found gdb(12.0.90) at /usr/bin/gdb.
[i 0522 20:50:45.523697 64 __init__.py:411] Found addr2line(2.38) at /usr/bin/addr2line.
[i 0522 20:50:45.642456 64 compiler.py:1010] cuda key:cu11.2.152_sm_61
Traceback (most recent call last):
File “test.py”, line 3, in
import torch
File “/home/xinjialong/anaconda3/envs/xin/lib/python3.7/site-packages/torch/init.py”, line 4, in
import jittor as jt
File “/home/xinjialong/anaconda3/envs/xin/lib/python3.7/site-packages/jittor/init.py”, line 18, in
from . import compiler
File “/home/xinjialong/anaconda3/envs/xin/lib/python3.7/site-packages/jittor/compiler.py”, line 1189, in
check_cache_compile()
File “/home/xinjialong/anaconda3/envs/xin/lib/python3.7/site-packages/jittor/compiler.py”, line 891, in check_cache_compile
assert jit_utils.cc
AssertionError