Companion

object Companion

Functions

Link copied to clipboard
fun getImageTag(cudaVersion: CUDAVersion, cudnn: Boolean): String

Generates the required image tag based on the requested cudaVersion and whether cudnn is needed.

Link copied to clipboard
fun getTag(cudaVersion: CUDAVersion, cudnn: Boolean): String