TensorFlow-gpu 安裝與 Anaconda

Question

這個周末我一直在嘗試安裝並獲得支持 GPU 的 Tensorflow以在我的計算機上工作，但我在使用 pip/conda 方面不是很有經驗，在觀看並嘗試了很多不同的教程/方法后，我現在很困惑web。

我有一個GeForce GTX 1650顯卡，我已經安裝了Cuda 10.0 （也是 11.2，但是我從“PATH”中刪除了它並且只使用了 10.0 版本，我認為這不是問題）。

我已經為 CUDA 10 下載了 cuDNN 7.5.0 ，並且我認為我已經正確復制並放置了文件（安裝了 cuDNN）。

我只是想讓一些版本的 Tensorflow-gpu 工作，但你可以在圖像上看到我現在一直在嘗試的 Tensorflow 版本。

I have tried to install and uninstall Python from my computer (I've also reinstalled Anaconda a lot of times), because I am not sure if I need to have a Python version installed (on my system) if I install a version of Python在我的 Anaconda 環境中（在我的示例中為 Python 3.7 ）。

Does anyone know how to install Tensorflow GPU on Windows 10 with my settings (cuDNN 7.5.0, CUDA 10), or maybe have encountered some trouble with Python versions or Anaconda problems similar to mine?

Answer 1

按照以下步驟在 windows 系統上安裝 Tensorflow GPU。

1. 確保安裝了正確版本的 Visual Studio。 在這里檢查。

2. 按照此處提到的說明為 windows 系統設置 CUDA

3. 安裝Tensorflow

 #check current python version python --version #Create the virtual environment conda create -n tf python=PYTHON_VERSION #Activate the tf environment conda activate tf #Install Tensorflow pip install tensorflow #Install CUDA and cuDNN using conda and make sure CUDA and cuDNN version should match the Tensorflow version conda install -c anaconda cudatoolkit=10.0 cudnn=7.5