自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(8)
  • 收藏
  • 关注

原创 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 284

原创 在 Linux 系统中加入 CUDA 环境变量

【代码】在 Linux 系统中加入 CUDA 环境变量。

2024-03-20 15:59:42 676 1

原创 Pytorch 计算模型的参数量

【代码】Pytorch 计算模型的参数量。

2023-11-15 11:03:54 150

原创 在linux服务器中创建新用户

【代码】在linux服务器中创建新用户。

2023-10-11 11:30:16 83

原创 在 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 221

原创 PyTorch查找张量中指定元素索引

【代码】PyTorch查找张量中指定元素索引。

2023-07-25 17:22:46 482

原创 vscode与Github同步文件

7. 将修改的文件push到仓库,分支“branch”是在执行完第六步之后默认创建的,名称用户可自己设定,一般默认为master。2. 在当前工程目录下撰写.gitignore文件,忽略不需要同步的文件。5. 将所有修改过的文件添加到staged changes。1. 在Github创建仓库,并复制仓库的网址。4. 连接Github仓库。6. 添加commit。

2023-06-24 16:25:39 471

原创 PyTorch 单机多卡DDP进行分布式训练

"nccl"指各GPU间的通信方式world_size参数指当前节点上有多少张卡rank指当前进程是在哪一张GPU卡上,其argument由torch.multiprocessing.spawn传入。

2023-05-27 09:57:24 1129 1

空空如也

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除