scnulpc的博客

1.cuda, pytorch版本对应问题 设备上的GPU驱动版本决定了cuda版本，而cuda版本与cudatoolkit的版本一致， 而pytorch（GPU）的版本要与cudatoolkit的版本严格一致。 一开始没留意这个版本的对应关系，直接输入官网上给出的命令，踩了大坑！（官网上的cudatoolkit版本为9.2，而我设备上的cudatoolkit版本为9.0，不...

官网说明： torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor) Returns the k largest elements of the given input tensor along a given dimension. If dim is not ...

官方解析： torch.Tensor.detach(): Returns a new Tensor, detached from the current graph. The result will never require gradient. 举个例子： 在GAN训练中，训练判别器D时，无需调整生成器G的参数。 fake_image = Generato...