
torch
文章平均质量分 62
且向长河问流水
这个作者很懒,什么都没留下…
展开
-
【PIL与Tensor】Pytorch中Tensor与PIL图像格式相互转化
首先,将Tensor复制到CPU并调整维度。然后使用.numpy()函数将Tensor转换为numpy数组,并乘以255以还原为原始图像数据类型。最后使用Image.fromarray()将numpy数组转换为PIL Image。其中,np.array()将PIL Image转换为numpy数组,.permute()调整了数组的维度以适应Pytorch Tensor的格式,并将数据类型转换为float类型。可以看到当前转化是正确的!原创 2023-05-22 10:02:06 · 2056 阅读 · 1 评论 -
[ncclUnhandledCudaError] unhandled cuda error, NCCL version xx.x.x
根据报错信息分析,分布式训练时在初始化的时候报错,并不是在训练的时候报错。因此,将问题定位到分布式训练初始化上。输入一下命令,检查当前服务器的卡。然而以上两种并没有解决问题。分布式训练的时候遇到问题。发现第一张卡是3070。于是,这里直接尝试使用。...原创 2022-08-16 15:22:01 · 6601 阅读 · 4 评论