logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

CUDA ERROR: device-side assert triggered at 问题及解决思路

cuda errorRuntimeError: cuda runtime error (59) : device-side assert triggered at ...我之前还以为是因为GPU抽风了引发的BUG,所以第一次没有在意,直接又重新开始运行了一次,但是第二次就发现程序在同样的地方断掉了,这也就想起来我以前看到的一个博客,里面有句话的大概意思是这样的:每次都在同样的地方出错的...

#深度学习#人工智能
如何在Pytorch中对Tensor进行shuffle

今天在训练网络的时候,考虑做一个实验需要将pytorch里面的某个Tensor沿着特征维度进行shuffle,之前考虑的是直接使用shuffle函数(random.shuffle),但是发现random.shuffle函数会导致Tensor里面数据被重复提取,进而导致输出跟输入分布发生变化。a = torch.randint(0, 2, (3, 3))print(a)random.shuffle(

#神经网络#pytorch#深度学习 +1
到底了