Summary: | sci-libs/tensorflow: ebuilds break configuration of CUDA compute capability level | ||
---|---|---|---|
Product: | Gentoo Linux | Reporter: | Soren Harward <stharward> |
Component: | Current packages | Assignee: | Jason Zaman <perfinion> |
Status: | RESOLVED FIXED | ||
Severity: | normal | ||
Priority: | Normal | ||
Version: | unspecified | ||
Hardware: | All | ||
OS: | Linux | ||
Whiteboard: | |||
Package list: | Runtime testing required: | --- |
Description
Soren Harward
2019-09-23 02:03:23 UTC
Update: I was wrong about #3. emerge will find and use the TF_CUDA_COMPUTE_CAPABILITIES variable if it is set in the shell. So maybe the best way to fix this bug is to have the ebuild display a warning during the configure stage if this environment variable is not set. The warning could be something like: WARNING: Tensorflow is being built with its default CUDA compute capabilities: 3.5 and 7.0. These may not be optimal for your GPU. To configure Tensorflow with the CUDA compute capability that is optimal for your GPU, set the environment variable TF_CUDA_COMPUTE_CAPABILITIES and then re-emerge tensorflow. For example, to use CUDA capability 7.5, run: $ TF_CUDA_COMPUTE_CAPABILITIES=7.5 emerge sci-libs/tensorflow You can look up your GPU's CUDA compute capability at https://developer.nvidia.com/cuda-gpus or by running $ /opt/cuda/extras/demo_suite/deviceQuery | grep "CUDA Capability" you can also just put TF_CUDA_COMPUTE_CAPABILITIES=7.5 in your make.conf, but yeah i should put a note in the ebuild. The bug has been closed via the following commit(s): https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=a3dc69074dcad86d4c95e024d231c90c62483152 commit a3dc69074dcad86d4c95e024d231c90c62483152 Author: Jason Zaman <perfinion@gentoo.org> AuthorDate: 2019-12-08 11:18:22 +0000 Commit: Jason Zaman <perfinion@gentoo.org> CommitDate: 2019-12-08 17:25:26 +0000 sci-libs/tensorflow: fix bazel, jsoncpp deps Also add a message about setting cuda compute capability Closes: https://bugs.gentoo.org/695428 Closes: https://bugs.gentoo.org/697864 Closes: https://bugs.gentoo.org/702222 Package-Manager: Portage-2.3.79, Repoman-2.3.16 Signed-off-by: Jason Zaman <perfinion@gentoo.org> sci-libs/tensorflow/Manifest | 1 + sci-libs/tensorflow/tensorflow-1.15.0_rc0.ebuild | 16 ++++++++++++++-- sci-libs/tensorflow/tensorflow-2.0.0.ebuild | 16 ++++++++++++++-- sci-libs/tensorflow/tensorflow-2.1.0_rc0.ebuild | 19 ++++++++++++++++--- 4 files changed, 45 insertions(+), 7 deletions(-) |