PyTorch
关注
分享
扫一扫
文章平均质量分 63
G.Chenhui
这个作者很懒，什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
【PyTorch】libtorch_cpu.so: undefined symbol: iJIT_NotifyEvent
这个符号在 MKL 2024.1 中被移除了，然而 PyTorch 是根据旧版本的 MKL 构建的。MKL 全称 Intel Math Kernel Library， 是由 Intel 公司开发的，专门用于矩阵计算的库。原创 2024-11-15 15:40:56 · 9439 阅读 · 8 评论 -
【PyTorch】DataLoader 设置 num_workers ＞ 0 时，出现 CUDA with multiprocessing 相关报错
Tensor 默认是在 CPU 上创建的，当我在数据集 Class 的。中 return 时，将 Tensor 转移到了 GPU 上。，在 return 时不将 Tensor 转移到 GPU 上。，导致数据在多进程加载时使用了 CUDA Tensor。同时，我在 DataLoader 定义时，设置了。修改数据集 Class 的。原创 2024-10-10 22:34:01 · 1359 阅读 · 0 评论