小编ans*_*ans的帖子

我一直在运行此笔记本，运行时类型为“高 RAM”“GPU”。我收到以下错误：

CUDA 内存不足。尝试分配 64.00 MiB（GPU 0；15.90 GiB 总容量；已分配 14.81 GiB；31.75 MiB 空闲；PyTorch 总共保留 14.94 GiB）

所以我从 Pro 升级到 Pro+，因为这应该会给我更多的内存，但我仍然遇到同样的错误。

我有以下功能。它适用于short：

void foo (short n, short * res)
{
  *res = n*2;
}

我想将结果存储在double变量中。例如：

short in = 5;
double out = 0;
foo(in,&out);

但这样的结果是一些垃圾。有没有办法在这种情况下强制转换这些类型，或者唯一的解决方案是使用一些类型的临时变量short？