如何解决slurm 集群上的 TensorFlow 2 仅检测到 1 个 CPU
我第一次使用 tensorflow(Python 3.8.6 中的 2.4.1)并且可以访问我大学使用 Slurm 的计算集群。我在一个节点上运行我的代码,该节点有 2 个 CPU,每个 CPU 有 20 个内核,但是当用 tf.config.list_physical_devices("CPU")
打印出 CPU 的数量时,我只得到 1。
我在网上搜索但找不到解决方案。我认为 tf 会默认使用所有可用的 CPU。我错过了什么吗?
我在 bash 脚本中使用的 slurm 参数是:
#SBATCH --ntasks=1
#SBATCH --time=10
#SBATCH --mem=10gb
#SBATCH --job-name=test
#SBATCH --nodes=1
#SBATCH --cpus-per-task=40
任何帮助和建议将不胜感激!
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。