我为 Geforce 2080 ti 安装了 Cuda 10.1 和最新的 Nvidia 驱动程序。我尝试运行一个基本脚本来测试 pytorch 是否正常工作，但出现以下错误：
RuntimeError: cuda runtime error (999) : unknown error at ..\aten\src\THC\THCGeneral.cpp:50
下面是我试图运行的代码：
import torch
torch.cuda.current_device()
torch.cuda.is_available()
torch.cuda.get_device_name(0)
我有一个这样的一维数据帧设置:
[A1,B1,C1,A2,B2,C2,A3,B3,C3,A4,B4,C4,A5,B5,C5,A6,B6,C6]
在我的程序A1,...中,C6将是从csv读取的数字.我想将它重塑为2d数据帧,如下所示:
[A1,B1,C1]
[A2,B2,C2]
[A3,B3,C3]
[A4,B4,C4]
[A5,B5,C5]
[A6,B6,C6]
我可以使用循环来制作它,但它会使程序减慢很多,因为我会多次进行这种转换.以这种方式重塑数据的最佳命令是什么？我查看了一堆重构数据帧问题,但找不到任何具体的内容.提前致谢.