cuda线程设置问题

tonqw 2020-03-31 06:53:34
dim3 grid=dim（1024,1,1）；dim3 bs=dim（1024,1,1）时没问题，但是把dim3 grid=dim（2048,1,1），最后的结果就都为0了，很奇怪？
...全文
600 回复 打赏 收藏 转发到动态 举报
写回复
用AI写文章
回复
切换为时间正序
请发表友善的回复…
发表回复
错误记录——CUDA程序运行结果不一致（线程块、网格大小设置问题）
使用GPU进行加速计算时，使用的线程索引数最好与需要计算的数据维度（一维长度、二维长宽）相等，这样才能保证计算结果的正确。
《CUDA并行程序设计-GPU编程指南》读书笔记--(1)线程网格、线程块以及线程
线程网格、线程块以及线程
python线程遇到问题，cuda遇到问题
报错：OpenBLAS warning: precompiled NUM_THREADS exceeded, adding auxiliary array for thread metadata.我主要是在命令行输入了export OPENBLAS_NUM_THREADS=1。程序开头加那个os的，我这个没有用处，还是在命令行输入好了。解决：搜索了很多博主大大的感觉这个博主的最有用。
完美解决pytorch多线程问题：Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing
在pytorch，dataloader里设置了num_workers > 1的时候, 会出现这个错误。 一般网上https://blog.csdn.net/qazwsxrx/article/details/116806358解决方法，加上一行设定。 然而，我这里如果加上这行代码，会导致代码卡死在dataloader里面。 后面发现，代码里面设定了这样的几行。 if cuda: torch.set_default_tensor_type('torch.cuda.FloatTen.
线程GPU id问题
主线程设置的GPU不会被继承，子线程会默认切换到0号GPU上。需要重新设置工作的GPU id。
CUDA on Linux

374

社区成员

345

社区内容

发帖
与我相关
我的任务
社区描述
CUDA on Linux
社区管理员
  • CUDA on Linux社区
加入社区
  • 近7日
  • 近30日
  • 至今

加载中

查看更多榜单
社区公告
暂无公告

试试用AI创作助手写篇文章吧

+ 用AI写文章