Fatal编程技术网
  • tensorflow/
  • Tensorflow ROCm和CUDA GPU在一个模型上？

Tensorflow ROCm和CUDA GPU在一个模型上？

tensorflow keras

Tensorflow ROCm和CUDA GPU在一个模型上？,tensorflow,keras,amd-rocm,Tensorflow,Keras,Amd Rocm,我想在多个GPU上编译一个模型。是否可以在同一型号上使用AMD GPU和ROCm以及Nvidia CUDA GPU？ 我想这是不可能的，因为您需要安装tensorflow的特殊版本才能启用ROCm tensorflow

我想在多个GPU上编译一个模型。是否可以在同一型号上使用AMD GPU和ROCm以及Nvidia CUDA GPU？ 我想这是不可能的，因为您需要安装tensorflow的特殊版本才能启用ROCm tensorflow