u010454261的博客

pytorch tensor 获取指定值的indices

tui jiahttps://www.zywvvd.com/2020/12/03/deep_learning/pytorch/dataloader_segmentation_fault/dataloader_segmentation_fault/

写在开头: 本文仅记录自己遇到的问题，方便之后查阅。最近用不同版本的pytorch训了一些模型，但使用pytorch1.3加载pytorch1.6训练的模型失败，显示如下所示等一些乱七八糟的错误。但是没关系，因为本宝想到了一个好的解决办法，hiahiahia~1. pytorch1.6 加载它自身生成的模型是没问题的，然后把加载后的模型dict生成pickle文件, 代码如下:importpickleaspkl...

今天评测时，总莫名出现GPU OOM的问题。参考https://cloud.tencent.com/developer/article/1626387， 发现了一种在变量使用完之后释放显存资源的方法.torch.cuda.empty_cache()

参考https://stackoverflow.com/questions/46826218/pytorch-how-to-get-the-shape-of-a-tensor-as-a-list-of-int>>> import torch>>> from torch.autograd import Variable>>> from ...

最近在使用pytorch写程序，想在指定的GPU设备上执行。假设函数func( )使用到了GPU设备：则在func()函数内部或者调用func()的地方加上:os.environ["CUDA_VISIBLE_DEVICES"] = str(gpu_id)我之前尝试在此处添加torch.cuda.set_device(self.gpu_id)，没作用，记录下。...