在 Anaconda 中安裝 tensorflow-gpu 時出現無法滿足的錯誤

Question

我已經在 Ubuntu 18.04 中安裝了 Anaconda Python 3.7，然后執行了以下命令：

conda update --all
conda install cudnn

現在，當我嘗試使用命令conda install tensorflow-gpu ，出現如下 Unsatisfiable 錯誤：

UnsatisfiableError：發現以下規范彼此不兼容：

• pkgs/main/linux-64::_ipyw_jlab_nb_ext_conf==0.1.0=py37_0 -> ipywidgets -> widgetsnbextension[version='>=3.4.0,<3.5.0'] -> notebook[version='>=4.4。 1'] -> nbconvert -> 漂白
• pkgs/main/linux-64::bleach==3.1.0=py37_0
• pkgs/main/linux-64::ipywidgets==7.4.2=py37_0 -> widgetsnbextension[version='>=3.4.0,<3.5.0'] -> notebook[version='>=4.4.1'] -> nbconvert -> 漂白
• pkgs/main/linux-64::jupyterlab==0.35.5=py37hf63ae98_0 -> jupyterlab_server[version='>=0.2.0,<0.3.0'] -> notebook -> nbconvert -> 漂白
• pkgs/main/linux-64::jupyterlab_server==0.2.0=py37_0 -> notebook -> nbconvert -> 漂白
• pkgs/main/linux-64::notebook==5.7.8=py37_0 -> nbconvert -> 漂白
• pkgs/main/linux-64::widgetsnbextension==3.4.2=py37_0 -> notebook[version='>=4.4.1'] -> nbconvert -> 漂白
• pkgs/main/noarch::nbconvert==5.5.0=py_0 -> 漂白

由於這里涉及多個重要的包，我很困惑該怎么做。 我的 PC 中裝有 NVIDIA GTX 1070 Max-Q，因此 tensorflow-gpu 應該可以完美運行。

Answer 1

您可以使用以下代碼安裝 Tensorflow GPU：

1. 創建新的虛擬環境

   conda create -n tensorflow_gpu pip python=3.6

2. 激活虛擬環境

   activate tensorflow_gpu

3. 安裝 CUDA Toolkit 使用

   conda install -c anaconda cudatoolkit

4. 安裝 Tensorflow GPU

   pip install --ignore-installed --upgrade tensorflow-gpu==1.15

有關更多信息，請參閱此Tensorflow 博客。