各位大佬好，最近在服务器上训练学生行为识别项目（目标检测 + VideoMAE动作识别），使用的是海光 Z100L GPU，目前卡在 PyTorch 环境兼容问题，求助一下。

## 服务器环境

GPU：
海光 Z100L

驱动：

hy-smi -a

输出：

Driver Version: 6.3.3-V1.8.0
Card Series: Z100L
Vendor: Chengdu Haiguang IC Design Co., Ltd.

说明 GPU 驱动正常。

---

## 当前 PyTorch 环境

当前安装：

torch 2.4.1+rocm6.1
torchvision 0.19.1+rocm6.1
torchaudio 2.4.1+rocm6.1

测试：

python -c "import torch; print(torch.cuda.is_available())"

返回：

True

python -c "import torch; print(torch.cuda.device_count())"

返回：

1

python -c "import torch; print(torch.cuda.get_device_name(0))"

返回：

AMD Radeon Graphics

---

## 当前问题

以下操作正常：

torch.cuda.is_available()
torch.cuda.init()
torch.empty(..., device="cuda")

但是以下操作直接 segfault：

x = torch.randn(3,3).cuda()
print(x+1)

或者：

import torch.nn as nn
m = nn.Linear(10,5).cuda()

以及：

train_videomae.py

都会直接 segmentation fault

---

## 已排查

我搜索到公网资料发现：

似乎需要海光 DTK 定制版：

torch-2.4.1+das.opt1.dtk25041-cp310-cp310-manylinux_2_28_x86_64.whl

并且看到有文章提到来源：

10.16.4.1:8000

但当前容器无法访问该内网地址。

---

## 想请教大家

1. 有没有人用过海光 Z100L？
2. 有没有这个 wheel 文件？
3. 是否有公网下载地址？
4. 是否可以用其他版本替代？

非常感谢！