Unity中如何创建可GPU写入的Texture3D？解决UAV标志报错

核心问题 #

你创建的Texture3D未启用随机写入（UAV）权限，Shader中的RWTexture3D要求纹理必须具备UAV使用标志，才能绑定为可写资源。

解决步骤 #

1. 创建支持UAV的Texture3D #

替换原有的new Texture3D(...)构造方式，使用Texture3D.Create方法启用随机写入权限，同时禁用CPU副本以实现“数据仅存于GPU”的需求：

// 创建8x8x8的RFloat格式Texture3D，启用UAV，禁用CPU访问
Texture3D tex = Texture3D.Create(
    width: 8,
    height: 8,
    depth: 8,
    format: TextureFormat.RFloat,
    mipCount: 1,
    enableRandomWrite: true, // 关键：开启UAV权限
    linear: false,
    createFlags: TextureCreationFlags.None,
    cpuAccessFlags: CPUAccessFlags.None // 禁止CPU读写，数据仅保留在GPU
);

2. 在GPU端完成纹理初始化（无需CPU副本） #

既然不想保留RAM中的数据，就不要用CPU数组初始化后上传，改用ComputeShader直接在GPU上完成初始数据填充：

• 先编写初始化用的ComputeShader（例如命名为InitTexture3D.compute）：

#pragma kernel InitTexture

RWTexture3D<float> points;

[numthreads(8,8,1)]
void InitTexture (uint3 id : SV_DispatchThreadID)
{
    bool isBorder = id.x == 0 || id.y == 0 || id.z == 0 || 
                    id.x == 7 || id.y == 7 || id.z == 7;
    points[id] = isBorder ? -1000.0f : 1.0f;
}

• 在C#中调用该ComputeShader完成初始化：

// 加载ComputeShader
ComputeShader initShader = Resources.Load<ComputeShader>("InitTexture3D");
// 获取Kernel索引
int kernelId = initShader.FindKernel("InitTexture");
// 绑定RWTexture3D
initShader.SetTexture(kernelId, "points", tex);
// 分发线程组：8x8x8纹理用1x1x8组，每组处理8x8x1个voxel
initShader.Dispatch(kernelId, 1, 1, 8);

3. 绑定到目标Shader并写入数据 #

现在Texture3D已具备UAV权限，可正常绑定到使用RWTexture3D<float>的Shader（以ComputeShader为例）：

ComputeShader targetShader = ...; // 你的目标计算Shader
int targetKernel = targetShader.FindKernel("YourTargetKernel");
targetShader.SetTexture(targetKernel, "points", tex);
targetShader.Dispatch(targetKernel, ...); // 根据纹理尺寸按需分发线程组

关键注意事项 #

• 确认纹理格式支持UAV：TextureFormat.RFloat是支持的，但部分压缩格式、低精度格式可能无法启用随机写入。
• 禁用CPU访问后，无法再通过SetPixelData/GetPixelData从CPU端读写纹理，所有操作必须通过GPU Shader完成。
• 不需要mipmap时，创建纹理将mipCount设为1，可减少GPU内存占用。

内容的提问来源于stack exchange，提问作者Jimmy Diddler