CompileOption

Neurocean小于 1 分钟

CompileOption

CompileOption Objects

class CompileOption()

编译选项

编译选择使用 Arguments 作为主要的存储载体,具有良好的二进制兼容性及扩展性

cpu

@staticmethod
def cpu()

cpu() -> ocean.ocean.CompileOption

CPU 默认编译选项

Returns:

编译选项

cuda

@staticmethod
def cuda()

cuda() -> ocean.ocean.CompileOption

CUDA 默认编译选项

Returns:

编译选项

setCudaDevices

def setCudaDevices(devices)

setCudaDevices(self: ocean.ocean.CompileOption, devices: str) -> None

设置CUDA设备

Parameter devices: 例如:

1. "0, 1, 2, 3"
2. "0"
3. "0,"
4. "all"

nodeArgs

参数载体

threadNumber

CPU线程数量

上次编辑于:
贡献者: damone