PythonWithCUDA

constructor(cudaVersion: CUDAVersion, cudnn: Boolean)

Parameters

cudaVersion

The requested version of CUDA.

cudnn

Whether to use a CUDA image which bundles cuDNN. The version of


constructor(cudaVersion: String, cudnn: Boolean)

Deprecated

Use the constructor accepting a CUDAVersion.