import jittor
from jittor.nn import UpsamplingBilinear2d, AvgPool2d
jittor.flags.use_cuda = 1 # 设为 0 可以
img = jittor.randn([1, 1, 2, 2])
img = AvgPool2d(2)(img)
print(img)
运行环境
jittor 1.3.5.18
cuda10.2_cudnn7_win
用 GPU运行出现一大堆看不明白的错误:
另外初学jittor 感觉有很多地方官方文档没有说明,比如
里说的那样,有些地方还要靠自己猜,我认为教程文档不够完善