在GPU上安装tensorflow

时间:2018-02-25 13:17:28

标签: python tensorflow gpu

我已经安装了tensorflow CPU版本。我使用的是Windows 10,我将AMD Radeon 8600M作为我的GPU。我现在可以安装GPU版本的tensorflow吗?会有问题吗?如果没有,我在哪里可以获得安装GPU版本的说明?

3 个答案:

答案 0 :(得分:0)

您的显卡不支持CUDA驱动程序,否则您无法在GPU上使用tensorflow。您的系统将运行tensorflow但仅在CPU上运行。

但是你可以使用pytorch它是另一种类似任务的方式。 PyTorch有另一个名为CLTorch的版本,它运行在你的显卡上运行的OpenCL上。

请点击此链接了解更多详情。 https://github.com/hughperkins/cltorch

答案 1 :(得分:0)

首先,如果你想看到性能提升,你应该拥有更好的GPU,其次,Tensorflow使用CUDA,它仅适用于CUDA能力为3.0或更高的NVidia GPU。如果您真的想深入学习,我建议您使用一些云服务,例如AWS或Google Cloud。

答案 2 :(得分:-1)

这取决于你的显卡,它必须是 nvidia ,你必须在你的系统和SO上安装相应的 cuda 版本。然后,您已在已安装的CUDA版本上安装了相应的cuDNN

步骤

  1. 安装NVIDIA 367驱动程序
  2. 安装CUDA 8.0
  3. 安装cuDNN 5.0
  4. 重新启动
  5. 使用上述配置
  6. 从bazel源安装tensorflow