我全新安装了 Ubuntu 18.04.3,我正在尝试安装 tensorflow-rocm(用于 AMD GPU)版本 1.14.0。
默认 pip3 install tensorflow-rocm 正在安装 v2.0 但我使用的代码集是在 1.14 上制作的,所以当我尝试在 v2.0 上运行相同的代码时出现一些错误,主要是因为包的移动方式。
所以我找到了 tensorflow-rocm v 1.14.0 的源代码,但是当我尝试构建它时,我遇到了一个错误。我不知道为什么。我检查了我的系统上是否安装了 rocm,并根据他们的官方网站安装了它。
我面临的错误如下:
Starting local Bazel server and connecting to it...
ERROR: Skipping '//tensorflow/tools/pip_package:build_pip_package': error loading package 'tensorflow/tools/pip_package': Encountered error while reading extension file 'rocm/build_defs.bzl': no such package '@local_config_rocm//rocm': Traceback (most recent call last):
File "/home/heyitsabi/tensorflow-upstream/third_party/gpus/rocm_configure.bzl", line 861
_create_local_rocm_repository(repository_ctx)
File "/home/heyitsabi/tensorflow-upstream/third_party/gpus/rocm_configure.bzl", line 682, in _create_local_rocm_repository
make_copy_dir_rule(repository_ctx, name = "rccl-inclu...", <2 more arguments>)
File "/home/heyitsabi/tensorflow-upstream/third_party/gpus/cuda_configure.bzl", line 923, in make_copy_dir_rule
_read_dir(repository_ctx, src_dir)
File "/home/heyitsabi/tensorflow-upstream/third_party/gpus/cuda_configure.bzl", line 956, in _read_dir
_execute(repository_ctx, ["find", src_dir, ..."], ...)
File "/home/heyitsabi/tensorflow-upstream/third_party/gpus/cuda_configure.bzl", line 887, in _execute
auto_configure_fail("\n".join([error_msg.strip() if ... ""]))
File "/home/heyitsabi/tensorflow-upstream/third_party/gpus/cuda_configure.bzl", line 324, in auto_configure_fail
fail(("\n%sCuda Configuration Error:%...)))
Cuda Configuration Error: Repository command failed
find: ‘/opt/rocm/rccl/include’: No such file or directory
WARNING: Target pattern parsing failed.
ERROR: error loading package 'tensorflow/tools/pip_package': Encountered error while reading extension file 'rocm/build_defs.bzl': no such package '@local_config_rocm//rocm': Traceback (most recent call last):
File "/home/heyitsabi/tensorflow-upstream/third_party/gpus/rocm_configure.bzl", line 861
_create_local_rocm_repository(repository_ctx)
File "/home/heyitsabi/tensorflow-upstream/third_party/gpus/rocm_configure.bzl", line 682, in _create_local_rocm_repository
make_copy_dir_rule(repository_ctx, name = "rccl-inclu...", <2 more arguments>)
File "/home/heyitsabi/tensorflow-upstream/third_party/gpus/cuda_configure.bzl", line 923, in make_copy_dir_rule
_read_dir(repository_ctx, src_dir)
File "/home/heyitsabi/tensorflow-upstream/third_party/gpus/cuda_configure.bzl", line 956, in _read_dir
_execute(repository_ctx, ["find", src_dir, ..."], ...)
File "/home/heyitsabi/tensorflow-upstream/third_party/gpus/cuda_configure.bzl", line 887, in _execute
auto_configure_fail("\n".join([error_msg.strip() if ... ""]))
File "/home/heyitsabi/tensorflow-upstream/third_party/gpus/cuda_configure.bzl", line 324, in auto_configure_fail
fail(("\n%sCuda Configuration Error:%...)))
Cuda Configuration Error: Repository command failed
find: ‘/opt/rocm/rccl/include’: No such file or directory
INFO: Elapsed time: 2.470s
INFO: 0 processes.
FAILED: Build did NOT complete successfully (0 packages loaded)
currently loading: tensorflow/tools/pip_package