-
以下是转换出错的代码:
pytorch_code="""
def conv_params(in_size, out_size):
filters = [3,2,5,4]
strides = [1,2,3] # max_stride = 3
pads = [0,1,2,3] # max pad
if out_size == 1:
…
-
```py
import jittor as jt
NUM_BATCHES = 10 # 如果有多个 gpu,尽量可以正好分开
BATCH_SIZE = 3
NUM_CLASSES = 3
def function_jittor():
jt.misc.set_global_seed(seed=986, different_seed_for_mpi=False)…
-
您好,看到您对不同框架进行测试,我们表示非常感谢,我运行了一下您的脚本,发现其中存在一个 小问题, jittor使用的dataset并没有使用多个worker,导致显卡其实没有跑满,我对您的代码进行了一行的改动,速度就大幅提升了,我在RTX TITAN上测cifar 10,一个epoch只需要48s,理论上,您的3090会更快。
```
train_loader.set_attrs(bat…
-
看起来是导入cuda出现了问题。**CUDA found but cublas is not loaded.**
`ImportError: /usr/local/cuda-11.1/lib64/libcublas.so.11: symbol free_gemm_select version libcublasLt.so.11 not defined in file libcublasLt.so…
-
When the pytorch version can be released? I am interested in the manifold part of this work, and looking forward to the code release.
-
-
所有都是即时的
Jittor的所有代码都是即时编译并且运行的,包括Jittor本身。用户可以随时对Jittor的所有代码进行修改,并且动态运行。
-
[i 0518 10:10:26.348652 20 cuda_flags.cc:39] CUDA enabled.
用户输入:1+1
Traceback (most recent call last):
File "cli_demo.py", line 9, in
model.chat()
File "/root/chatglm/JittorLLMs/models/c…
-
先说一下我的环境,win10,conda环境安装的python 3.10.11,根据requirements安装的包如下
pip list
Package Version
------------------ --------
astunparse 1.6.3
certifi 2023.5.7
charset-normali…
-
## Describe the bug
Fail to use **nn._fft2()**
After **nn._fft2()**, the data type has been changed from **jt.Var** to **jt.jittor_core.Var**, the log shows some errors during compiling.
## Full Lo…