docker-tensorflow-builder icon indicating copy to clipboard operation
docker-tensorflow-builder copied to clipboard
verified publisher icon hadim

compute compatibility

Open fennecinspace opened this issue 6 years ago • 1 comments

how do i specify the compute compatibility .. i would like to build with 3.0

fennecinspace avatar Nov 18 '19 22:11 fennecinspace

Hi,

in the build.sh script, located in the subfolder of the OS you want to build for, there is a:

export TF_CUDA_COMPUTE_CAPABILITIES=5.2,3.5

you just need to set this accordingly.

jeremyfix avatar Nov 22 '19 10:11 jeremyfix