- 博客(9)
- 收藏
- 关注
原创 Linux 中使用gcc, g++, nvcc 编译 PyTorch 扩展时,#include “torch/extension.h“ No such path or file.
【代码】Linux 中使用gcc, g++, nvcc 编译 PyTorch 扩展时,#include “torch/extension.h“ No such path or file.
2024-03-20 16:31:08
611
1
原创 在 import OpenEXR 过程中报错 OpenEXR.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZTIN13IlmThread_3
【代码】在 import OpenEXR 过程中报错 OpenEXR.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZTIN13IlmThread_3。
2023-10-03 09:15:40
307
1
原创 vscode与Github同步文件
7. 将修改的文件push到仓库,分支“branch”是在执行完第六步之后默认创建的,名称用户可自己设定,一般默认为master。2. 在当前工程目录下撰写.gitignore文件,忽略不需要同步的文件。5. 将所有修改过的文件添加到staged changes。1. 在Github创建仓库,并复制仓库的网址。4. 连接Github仓库。6. 添加commit。
2023-06-24 16:25:39
604
原创 PyTorch 单机多卡DDP进行分布式训练
"nccl"指各GPU间的通信方式world_size参数指当前节点上有多少张卡rank指当前进程是在哪一张GPU卡上,其argument由torch.multiprocessing.spawn传入。
2023-05-27 09:57:24
1323
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人