1. 首頁
  2. 問答
  3. Tensorflow與GPU和CUDA v5.5

Tensorflow與GPU和CUDA v5.5

2016-06-26 235 views
0

我想使用GPU啓用版本的Tensorflow。我讀過它與CUDA 7.5的最佳配合，但我使用的服務器安裝了CUDA 5.5版本。Tensorflow與GPU和CUDA v5.5

我可以用CUDA 5.5配置Tensorflow嗎？如果是，如何？我已經在一個虛擬環境中安裝了tensorflow。

來源

2016-06-26 pranav shukla

回答

1

從安裝文檔：

爲了建立或GPU支持運行TensorFlow，無論NVIDIA的CUDA工具包（> = 7.0）和cuDNN（> = V2）需要安裝。

如果您不從源代碼安裝，您將需要使用7.5版本。

另外，還要確保你有一個兼容GPU：

TensorFlow GPU的支持需要具有GPU卡的NVidia計算能力> = 3.0。支持的卡包括但不限於：

  • 的NVidia泰坦
  • 的NVidia泰坦X
  • 的NVidia K20
  • 的NVidia K40

來源

2016-06-26 21:10:31

相關問題

 相關問題