在Rocm中使用tensoreflow

时间:2019-06-14 20:55:46

标签: tensorflow amd-processor

我正在使用ROCM(Ubuntu HP envy x360 Ryzen 5 2500u)在我的AMD(RavenRidge APU)笔记本电脑上设置基于张量流的机器学习工作区

我已经安装了rocm并尝试使用基本的tensorflow脚本,但是出现了HIP错误。这是否意味着我的GPU与rocm不兼容?

2019-06-14 21:51:59.142628: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1189] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 7538 MB memory) -> physical GPU (device: 0, name: AMD Ryzen 5 2500U with Radeon Vega Mobile Gfx, pci bus id: 0000:03:00.0)
2019-06-14 21:51:59.221889: E tensorflow/stream_executor/rocm/rocm_driver.cc:601] failed to allocate 7.36G (7905050624 bytes) from device: hipError_t(1002)
2019-06-14 21:51:59.221972: E tensorflow/stream_executor/rocm/rocm_driver.cc:601] failed to allocate 6.62G (7114545152 bytes) from device: hipError_t(1002)
2019-06-14 21:51:59.221997: E tensorflow/stream_executor/rocm/rocm_driver.cc:601] failed to allocate 5.96G (6403090432 bytes) from device: hipError_t(1002)
2019-06-14 21:51:59.222021: E tensorflow/stream_executor/rocm/rocm_driver.cc:601] failed to allocate 5.37G (5762781184 bytes) from device: hipError_t(1002)
2019-06-14 21:51:59.222054: E tensorflow/stream_executor/rocm/rocm_driver.cc:601] failed to allocate 4.83G (5186503168 bytes) from device: hipError_t(1002)
2019-06-14 21:51:59.222090: E tensorflow/stream_executor/rocm/rocm_driver.cc:601] failed to allocate 4.35G (4667852800 bytes) from device: hipError_t(1002)
2019-06-14 21:52:05.830021: E tensorflow/stream_executor/rocm/rocm_driver.cc:461] failed to memset memory: HIP_ERROR_InvalidValue

2 个答案:

答案 0 :(得分:0)

我在Ryzen 5 2400G上遇到了同样的错误。这是讨论该问题的github线程。

https://github.com/ROCmSoftwarePlatform/tensorflow-upstream/issues/277

从评论#2

  

嗨安德里亚斯。 ROCm堆栈中的各种组件目前不支持我们的APU。可以在以下位置找到有关硬件支持的详细信息:https://github.com/RadeonOpenCompute/ROCm#supported-gpus

答案 1 :(得分:-1)

由于主板的BIOS,恐怕ROCm-HIP当前不受完全支持。

plz看到这个。

https://github.com/RadeonOpenCompute/ROCm/wiki#not-supported-or-limited-support-under-rocm