0
点赞
收藏
分享

微信扫一扫

nitialization error CUDA kernel errors CUDA_LAUNCH_BLOCKING=1 Compile with `TORCH_USE_CUDA_DSA`


CUDA error: initialization error CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1 Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.

原因是 pytorch   

torchDataset(x)
 x传入的不是list,而是tensor

 X_test =...... 
# X_test = torch.tensor(X_test, dtype=torch.float32).to(device)
# y_test = torch.tensor(y_test, dtype=torch.float32).to(device)

改成list就没有这个问题

举报

相关推荐

0 条评论