Jittor

这个是pytorch的问题，您试试安装pytorch1.8就能解决问题，如果还不行加上环境变量use_mkl=0 ---Original--- From: "Zihan ***@***.***> Date: Mon, May 31, 2021 16:47 PM To: ***@***.***>; Cc: ***@***.******@***.***>; Subject: Re: [Jittor/jittor] Support for LSTM, GRU, Conv3d, ConstantPad3d (#223) 我使用的是pytorch1.6.0+CUDA10.1。我使用jt.load()读取我在pytorch上的预训练模型的时候出现了以下的问题，main_model->video_model里面的jt.load()函数可以正常读取模型，但是在main_model进行jt.load()的时候报出munmap_chunk(): invalid pointer的错误。 —...

.numpy()的时候说明框架正在计算，因为jittor是延迟执行的，之前构建的计算图在.numpy的时候才会执行，卡一下是正常的，如果卡了很久可能是没有开启GPU加速，或者数据规模过大，如果您用了conv3d的话，目前conv3d还没有merge进主分支，速度较慢，推荐您把数据改小。 conv3d我们正在优化，很快会有个正式版本 ---Original--- From: "Zihan ***@***.***> Date: Mon, May 31, 2021 21:12 PM To: ***@***.***>; Cc: ***@***.******@***.***>; Subject: Re: [Jittor/jittor] Support for LSTM, GRU, Conv3d, ConstantPad3d (#223) 我改了pytorch1.8之后detach()可以正常执行了，但是numpy()又卡住，而且我改了环境变量还不行 — You...

这个应该是因为内存不足了，因为xx这些中间变量因为会被优化掉，但是如果打印出来的话，就会导致无法进行优化，从而内存不足 ---Original--- From: "Zihan ***@***.***> Date: Tue, Jun 1, 2021 14:25 PM To: ***@***.***>; Cc: ***@***.******@***.***>; Subject: Re: [Jittor/jittor] Support for LSTM, GRU, Conv3d, ConstantPad3d (#223) 发现新的bug，对于Conv2d函数，当xx.shape = [2,2048,2056,10,10,3,3,]时，会报出如下错误： 错误处代码位于nn.py...

这个看起来似乎是cudnn报错了，您的cuda版本和cudnn版本是多少呢，请问python3.7 -m jittor.test.test_resnet能成功运行吗？ ---Original--- From: "Zihan ***@***.***> Date: Tue, Jun 1, 2021 14:23 PM To: ***@***.***>; Cc: ***@***.******@***.***>; Subject: Re: [Jittor/jittor] Support for LSTM, GRU, Conv3d, ConstantPad3d (#223) 在我打开了GPU加速之后遇到这这种问题，这可能是什么原因呢？ 我使用的是anaconda中的虚拟环境...

conv3d 和 pool3d目前已经在主分支支持。

您的nvcc在什么位置呢，which nvcc输出怎样

建议您让jittor帮您找到合适版本的nvcc和cuda，您可以运行python3 -m jittor_utils.install_cuda

错误信息里面显示`Exception: Run cmd failed: /usr/loca/cuda/bin/nvcc --version`， 您的环境变量是不是`local`误写成`loca`了？

代码1的`DistributedDataParallel`，在jittor中是不需要的，jittor的分布式更加简单，直接删掉即可 identity， UpsamplingBilinear2d， AdaptiveMaxPool2d已经在最新的1.2.2.71版本加上啦，您可以试试

感谢您对计图的关注，ONNX支持我们正在开发中，计图自带的元算子融合加速可以替代TensorRT进行加速。