- 博客(3)
- 收藏
- 关注
RSS订阅原创 全连接层居然可以处理三维数据
今天发现一个有意思的地方输入shape为：torch.Size([1, 8, 64])全连接层： nn.Linear(64, 1)居然没有报错，而且输出为torch.Size([1, 8, 1])！nb
2021-01-06 16:54:56
3318
6
原创 CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
昨天跑的程序，今天就跑不了。在cpu上跑就可以，在GPU上就报错CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)最后解决：更新nvidia显卡驱动，重启电脑。
2021-01-06 16:38:04
5225
3
原创 更新驱动之后CUDA error: unknown error
（已解决）更新驱动之后CUDA error: unknown error重启电脑！！！！！！！！尝试n种方法，最后重启搞定。
2021-01-06 16:32:54
1958
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人