为什么在conda安装后Tensorflow无法识别我的GPU?

30
我是深度学习的新手,过去两天一直在尝试在我的电脑上安装tensorflow-gpu版本但都没有成功。由于在线论坛中存在许多兼容性问题,我避免安装CUDA和cuDNN驱动程序。既然我之前已经使用conda分发的Python,因此我选择按照官方网站这里写的conda install -c anaconda tensorflow-gpu进行安装:https://anaconda.org/anaconda/tensorflow-gpu

然而,即使我在一个全新的虚拟环境中安装了gpu版本(以避免与基础环境中安装的pip库发生冲突),tensorflow似乎出于某种神秘的原因甚至都没有识别到我的GPU。

我运行了一些代码片段(在Anaconda提示符中)来了解它为什么无法识别我的GPU:

1.

>>>from tensorflow.python.client import device_lib
        >>>print(device_lib.list_local_devices())
                    [name: "/device:CPU:0"
                device_type: "CPU"
                memory_limit: 268435456
                locality {
                }
                incarnation: 7692219132769779763
                ]

正如你所看到的,它完全忽略了GPU。

2.

>>>tf.debugging.set_log_device_placement(True)
    >>>a = tf.constant([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
2020-12-13 10:11:30.902956: I tensorflow/core/platform/cpu_feature_guard.cc:142] This 
TensorFlow 
binary is optimized with oneAPI Deep Neural Network Library (oneDNN)to use the following CPU 
instructions in performance-critical operations:  AVX AVX2
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
>>>b = tf.constant([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]])
>>>c = tf.matmul(a, b)
>>>print(c)
tf.Tensor(
[[22. 28.]
[49. 64.]], shape=(2, 2), dtype=float32)

这里本应该通过显示执行op MatMul,设备为/job:localhost/replica:0/task:0/device:GPU:0(如此写在这里:https://www.tensorflow.org/guide/gpu)来表明它是在GPU上运行的,但没有任何类似的情况出现。另外我不确定第二行之后的信息是什么意思。

我也在网上搜索了几个解决方案,包括这里,但几乎所有的问题都与第一种手动安装方法有关,而我还没有尝试过,因为每个人都推荐使用这种方法。

我不再使用cmd,因为在从基本环境中卸载tensorflow-cpu并重新安装之后,环境变量似乎出了点问题。用anaconda提示符完美地工作了,但cmd却不行。这是一个单独的问题(也很普遍),但我提到它是因为它可能在这里扮演了一个角色。我在一个全新的虚拟环境中安装了gpu版本,以确保干净的安装,就我所知,CUDA和cuDNN库的手动安装只需要设置路径变量。

我使用的显卡是:(支持CUDA)

C:\WINDOWS\system32>wmic path win32_VideoController get name
Name
NVIDIA GeForce 940MX
Intel(R) HD Graphics 620

我目前正在使用的Tensorflow和Python版本:

>>> import tensorflow as tf
>>> tf.__version__
'2.3.0'

Python 3.8.5 (default, Sep  3 2020, 21:29:08) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.

系统信息:Windows 10 Home,64位操作系统,基于x64的处理器。

非常感谢您的帮助。提前致谢。


抱歉格式不好,StackOverflow 不允许我在没有“缩进”终端代码的情况下发布问题。 - Sarosij Bose
1
我遇到了同样的问题。当我执行 conda list 命令时,发现 cudatoolkit 并没有被安装,尽管 tensorflow-gpu 已经安装好了。我还发现 cudnn 也没有安装。我正在尝试使用 conda install 命令来安装它们。当我执行 conda install cudnn 命令时,它要求我将刚刚安装的 cudatoolkit 11 降级为 cudatoolkit 10。也许 conda create -n tf-gpu tensorflow-gpu 在最近的版本(如 cudatoolkit 11)中效果不佳。 - user3731622
@user3731622 是的,我也不得不降级。看起来conda还没有为Tf和CUDA的新库提供支持。 - Sarosij Bose
我能够使用conda安装TensorFlow 2.1,然后使用pip安装TensorFlow 2.4。虽然我不喜欢混合使用conda和pip,但这是让新版本的TensorFlow运行的一种方法。 - user3731622
11个回答

51

2021年8月,根据下面评论区的@ComputerScientist所说,conda install tensorflow-gpu==2.4.1 可能已经可以使用了,会安装和

以下内容为2021年1月编写,已经过时

当前conda install tensorflow-gpu会安装tensorflow v2.3.0,并且不会安装conda cudnn或cudatoolkit软件包。手动安装它们(例如,conda install cudatoolkit=10.1)似乎也不能解决问题。

一种解决方法是安装较早版本的tensorflow,该版本会安装cudnn和cudatoolkit,然后通过pip升级。

conda install tensorflow-gpu=2.1
pip install tensorflow-gpu==2.3.1

2.4.0使用的是cuda 11.0和cudnn 8.0,但截至2020年12月16日,anaconda中尚未包含cudnn 8.0。

编辑:请参阅@GZ0的答案,其中链接到具有一行解决方案的github讨论。


2
谢谢@geometrikal,这对我很有帮助! - Sarosij Bose
1
一直在尝试解决这个问题。谢谢! - Edward
2
@TusharAgarwal 尝试使用conda create -n tf_231 python=3.7创建环境,然后使用conda activate tf_231激活环境;安装tensorflow-gpu=2.1请使用conda install,安装tensorflow-gpu==2.3.1请使用pip install。注意,“-gpu”可能不需要。 - geometrikal
2
@TusharAgarwal 请参考此帖子中的解决方法。 - GZ0
3
可能已经过时了。截至2021年8月,当我在新的Ubuntu 18.04机器和新的conda环境上执行conda install tensorflow-gpu==2.4.1时,会在我的conda环境中安装cudatoolkit-10.1.243cudnn-7.6.5。请注意,此回答仅供参考,可能已经不适用于当前日期。 - ComputerScientist
显示剩余7条评论

43
在安装tensorflow-gpu 2.3时,Anaconda在Windows 10上自动选择的tensorflow构建似乎存在问题。请参考此处的解决方法(如果您有GitHub帐户,请考虑点赞该GitHub答案)。 仅适用于Windows: Python 3.7: conda install tensorflow-gpu=2.3 tensorflow=2.3=mkl_py37h936c3e2_0 Python 3.8: conda install tensorflow-gpu=2.3 tensorflow=2.3=mkl_py38h1fcfbd6_0

请修复我的问题,谢谢。 - Yiling Liu
在第15页的中间,我瞥见了TL;DR,然后发现这个东西居然能用?太棒了! - J B
它在Windows 10上运行正常,但在Ubuntu上2.3不可用。谢谢! - Macumbaomuerte
同时令人惊叹和悲伤的。 - t0b4cc0

6

@geometrikal的解决方案对我几乎有用。但是在使用conda安装tensorflow-gpu和使用pip安装tensorflow 2.3之间,我需要卸载tensorflow-gpu包中的tensorflow部分,以避免pip出现一致性警告。conda会卸载整个包。我知道Conda不推荐混合使用pip和conda,但这是有效的解决方案,而且我已经厌倦了再花一天时间来解决这个问题。

conda create -n tfgpu python=3.7
conda activate tfgpu
conda install tensorflow-gpu=2.1

pip uninstall tensorflow
pip uninstall tensorflow-estimator
pip uninstall tensorboard 
pip uninstall tensorboard-plugin-wit
pip install tensorflow==2.3
pip check

这个问题是spyder、jupyter等软件安装在基础环境中,无法识别tensorflow的安装。现在我不能在新的tensorflow环境中重新安装每个软件包。 - samarendra chandan bindu Dash
@samarendrachandanbinduDash 对不起,我没有使用spyder或jupyter的经验。我只使用Pycharm和conda组合。 - tobi.tobt

5

我也无法(目前)让TF 2.3.0识别我的Nvidia Quadro Pro 620 GPU。

注意:我在这台PC上还有其他两个“环境”(Windows Pro),都是通过Anaconda安装的:

  1. Python 3.7.8 TF 2.0.0... 可以识别(且使用)Nvidia GPU
  2. Python 3.6.9 TF 2.1.0... 可以识别(且使用)Nvidia GPU
  3. Python 3.8.6 TF 2.3.0... 无法看到GPU

我的机器有Cuda 11.1; cuDNN 8.0.5

我的下一个想法是考虑将第三种配置中的Python从3.8.6降级到3.7.8,其中TF = 2.3.0

Steve


3

如需使用GPU,您需要安装cuDNN和CUDA工具包。

首先,在此处检查兼容版本 here

cuDNN可以在此处找到(需要免费账户)

CUDA工具包可以在此处找到

再次强调,在安装之前请检查兼容版本。 新版本不向后兼容。


但是当我使用conda安装tensorflow-gpu时,这些不是自动安装的吗? - Sarosij Bose
这些是不特定于Python的低级库。 - Royce Schultz
1
是的,没错。但我在这里看到不需要外部安装:https://towardsdatascience.com/tensorflow-gpu-installation-made-easy-use-conda-instead-of-pip-52e5249374bc - Sarosij Bose
我不使用conda,所以不确定,但它确实说“安装tensorflow-gpu所需的所有软件包,包括cuda和cuDNN 兼容版本”,而不是cuda或cuDNN本身。这可能是一种管理先前安装的非GPU软件包的方法。 - Royce Schultz
哎呀,在软件包列表中确实包含cuDNN和CUDA。你有激活你的环境吗? - Royce Schultz
3
错误的。Conda将从CUDA工具包和CNN套件安装所有必要的运行时组件。它不会安装受支持的NVIDIA GPU驱动程序。 - talonmies

1
我看到你的GPU具有计算能力5.0,这很好,TensorFlow应该会喜欢它。因此,我认为在环境设置过程中出了问题。请尝试使用以下命令创建一个新的环境:
conda create --name tf_gpu tensorflow-gpu 

然后在tf_gpu中安装您想要的所有其他软件包,然后再试一次。

附言:确保在环境中只有一个TensorFlow软件包(GPU版本),如果有多个,则不能保证。

import tensorflow as tf

将导入您想要的那个...

0
在我的情况下(2022年4月):
 conda install tensorflow-gpu=2.3 tensorflow=2.3=mkl_py38h1fcfbd6_0 cudatoolkit cudnn keras matplotlib

完美运行!它安装了tensorflow-gpu=2.3 - cudatoolkit 10.1.243和cudnn 7.6.5


0
截至2021年8月,使用TensorFlow 2.4.1,我认为它似乎可以在conda环境中安装CUDA和CuDNN。以下是我在Ubuntu 18.04机器上创建新的conda环境所做的操作:
conda create --name tftest python=3.7 -y  &&  conda activate tftest
conda install ipython tensorflow-gpu==2.4.1 -y

上述命令将列出以下要安装的软件包。 对于我们的目的,请注意 cudatoolkit 和 cudnn 如何列出。
The following packages will be downloaded:

    package                    |            build
    ---------------------------|-----------------
    _tflow_select-2.1.0        |              gpu           2 KB
    absl-py-0.13.0             |   py37h06a4308_0         173 KB
    aiohttp-3.7.4              |   py37h27cfd23_1         536 KB
    astor-0.8.1                |   py37h06a4308_0          47 KB
    astunparse-1.6.3           |             py_0          17 KB
    async-timeout-3.0.1        |   py37h06a4308_0          13 KB
    attrs-21.2.0               |     pyhd3eb1b0_0          46 KB
    backcall-0.2.0             |     pyhd3eb1b0_0          13 KB
    blas-1.0                   |              mkl           6 KB
    blinker-1.4                |   py37h06a4308_0          23 KB
    brotlipy-0.7.0             |py37h27cfd23_1003         320 KB
    c-ares-1.17.1              |       h27cfd23_0         108 KB
    cachetools-4.2.2           |     pyhd3eb1b0_0          13 KB
    cffi-1.14.6                |   py37h400218f_0         223 KB
    chardet-3.0.4              |py37h06a4308_1003         175 KB
    charset-normalizer-2.0.4   |     pyhd3eb1b0_0          35 KB
    click-8.0.1                |     pyhd3eb1b0_0          79 KB
    coverage-5.5               |   py37h27cfd23_2         254 KB
    cryptography-3.4.7         |   py37hd23ed53_0         904 KB
    cudatoolkit-10.1.243       |       h6bb024c_0       347.4 MB
    cudnn-7.6.5                |       cuda10.1_0       179.9 MB
    cupti-10.1.168             |                0         1.4 MB
    cython-0.29.24             |   py37h295c915_0         1.9 MB
    decorator-5.0.9            |     pyhd3eb1b0_0          12 KB
    gast-0.4.0                 |     pyhd3eb1b0_0          13 KB
    google-auth-1.33.0         |     pyhd3eb1b0_0          80 KB
    google-auth-oauthlib-0.4.4 |     pyhd3eb1b0_0          18 KB
    google-pasta-0.2.0         |             py_0          46 KB
    grpcio-1.36.1              |   py37h2157cd5_1         1.9 MB
    h5py-2.10.0                |   py37hd6299e0_1         902 KB
    hdf5-1.10.6                |       hb1b8bf9_0         3.7 MB
    idna-3.2                   |     pyhd3eb1b0_0          48 KB
    importlib-metadata-3.10.0  |   py37h06a4308_0          33 KB
    intel-openmp-2021.3.0      |    h06a4308_3350         1.4 MB
    ipython-7.26.0             |   py37hb070fc8_0        1005 KB
    ipython_genutils-0.2.0     |     pyhd3eb1b0_1          27 KB
    jedi-0.18.0                |   py37h06a4308_1         911 KB
    keras-preprocessing-1.1.2  |     pyhd3eb1b0_0          35 KB
    libgfortran-ng-7.5.0       |      ha8ba4b0_17          22 KB
    libgfortran4-7.5.0         |      ha8ba4b0_17         995 KB
    libprotobuf-3.17.2         |       h4ff587b_1         2.0 MB
    markdown-3.3.4             |   py37h06a4308_0         127 KB
    matplotlib-inline-0.1.2    |     pyhd3eb1b0_2          12 KB
    mkl-2021.3.0               |     h06a4308_520       141.2 MB
    mkl-service-2.4.0          |   py37h7f8727e_0          56 KB
    mkl_fft-1.3.0              |   py37h42c9631_2         170 KB
    mkl_random-1.2.2           |   py37h51133e4_0         287 KB
    multidict-5.1.0            |   py37h27cfd23_2          66 KB
    numpy-1.20.3               |   py37hf144106_0          23 KB
    numpy-base-1.20.3          |   py37h74d4b33_0         4.5 MB
    oauthlib-3.1.1             |     pyhd3eb1b0_0          90 KB
    opt_einsum-3.3.0           |     pyhd3eb1b0_1          57 KB
    parso-0.8.2                |     pyhd3eb1b0_0          69 KB
    pexpect-4.8.0              |     pyhd3eb1b0_3          53 KB
    pickleshare-0.7.5          |  pyhd3eb1b0_1003          13 KB
    prompt-toolkit-3.0.17      |     pyh06a4308_0         256 KB
    protobuf-3.17.2            |   py37h295c915_0         319 KB
    ptyprocess-0.7.0           |     pyhd3eb1b0_2          17 KB
    pyasn1-0.4.8               |             py_0          57 KB
    pyasn1-modules-0.2.8       |             py_0          72 KB
    pygments-2.10.0            |     pyhd3eb1b0_0         725 KB
    pyjwt-2.1.0                |   py37h06a4308_0          32 KB
    pyopenssl-20.0.1           |     pyhd3eb1b0_1          49 KB
    pysocks-1.7.1              |           py37_1          27 KB
    python-flatbuffers-1.12    |     pyhd3eb1b0_0          24 KB
    requests-2.26.0            |     pyhd3eb1b0_0          59 KB
    requests-oauthlib-1.3.0    |             py_0          23 KB
    rsa-4.7.2                  |     pyhd3eb1b0_1          28 KB
    scipy-1.6.2                |   py37had2a1c9_1        15.5 MB
    six-1.16.0                 |     pyhd3eb1b0_0          18 KB
    tensorboard-2.4.0          |     pyhc547734_0         8.8 MB
    tensorboard-plugin-wit-1.6.0|             py_0         630 KB
    tensorflow-2.4.1           |gpu_py37ha2e99fa_0           4 KB
    tensorflow-base-2.4.1      |gpu_py37h29c2da4_0       195.2 MB
    tensorflow-estimator-2.5.0 |     pyh7b7c402_0         267 KB
    tensorflow-gpu-2.4.1       |       h30adc30_0           3 KB
    termcolor-1.1.0            |   py37h06a4308_1           9 KB
    traitlets-5.0.5            |     pyhd3eb1b0_0          81 KB
    typing-extensions-3.10.0.0 |       hd3eb1b0_0           8 KB
    typing_extensions-3.10.0.0 |     pyh06a4308_0          27 KB
    urllib3-1.26.6             |     pyhd3eb1b0_1         112 KB
    wcwidth-0.2.5              |             py_0          29 KB
    werkzeug-1.0.1             |     pyhd3eb1b0_0         239 KB
    wrapt-1.12.1               |   py37h7b6447c_1          49 KB
    yarl-1.6.3                 |   py37h27cfd23_0         133 KB
    zipp-3.5.0                 |     pyhd3eb1b0_0          13 KB
    ------------------------------------------------------------
                                           Total:       915.9 MB

运行 ipython 并尝试:
In [1]: import tensorflow as tf
2021-08-29 12:26:36.582384: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.10.1

In [2]: tf.config.list_physical_devices('GPU')
2021-08-29 12:26:48.676151: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2021-08-29 12:26:48.679894: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2021-08-29 12:26:48.975002: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:04:00.0 name: Tesla V100-PCIE-32GB computeCapability: 7.0
coreClock: 1.38GHz coreCount: 80 deviceMemorySize: 31.75GiB deviceMemoryBandwidth: 836.37GiB/s
2021-08-29 12:26:48.979341: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 1 with properties: 
pciBusID: 0000:08:00.0 name: Tesla V100-PCIE-32GB computeCapability: 7.0
coreClock: 1.38GHz coreCount: 80 deviceMemorySize: 31.75GiB deviceMemoryBandwidth: 836.37GiB/s
2021-08-29 12:26:48.981747: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 2 with properties: 
pciBusID: 0000:09:00.0 name: Tesla V100-PCIE-32GB computeCapability: 7.0
coreClock: 1.38GHz coreCount: 80 deviceMemorySize: 31.75GiB deviceMemoryBandwidth: 836.37GiB/s
2021-08-29 12:26:48.990002: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 3 with properties: 
pciBusID: 0000:85:00.0 name: Tesla V100-PCIE-32GB computeCapability: 7.0
coreClock: 1.38GHz coreCount: 80 deviceMemorySize: 31.75GiB deviceMemoryBandwidth: 836.37GiB/s
2021-08-29 12:26:48.992488: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 4 with properties: 
pciBusID: 0000:89:00.0 name: Tesla V100-PCIE-32GB computeCapability: 7.0
coreClock: 1.38GHz coreCount: 80 deviceMemorySize: 31.75GiB deviceMemoryBandwidth: 836.37GiB/s
2021-08-29 12:26:48.992523: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.10.1
2021-08-29 12:26:49.312793: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.10
2021-08-29 12:26:49.312907: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.10
2021-08-29 12:26:49.388961: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2021-08-29 12:26:49.413946: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2021-08-29 12:26:49.535055: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2021-08-29 12:26:49.563142: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.10
2021-08-29 12:26:50.009291: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.7
2021-08-29 12:26:50.051301: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0, 1, 2, 3, 4
Out[2]: 
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU'),
 PhysicalDevice(name='/physical_device:GPU:1', device_type='GPU'),
 PhysicalDevice(name='/physical_device:GPU:2', device_type='GPU'),
 PhysicalDevice(name='/physical_device:GPU:3', device_type='GPU'),
 PhysicalDevice(name='/physical_device:GPU:4', device_type='GPU')]

In [3]: tf.test.is_built_with_cuda()
Out[3]: True

这台机器有5个GPU,所以上面的输出是正确的。

我不确定的是为什么安装了CUDA 10.1和CuDNN 7.6.5,因为根据Google的TF兼容性图表,2.4.0(以及可能的2.4.1?)可以与CUDA 11.0和CuDNN 8一起使用。如果有人对此有见解,请随时发言...


很容易,TF 2.4.0是由Tensorflow开发人员使用CUDA 11.0和cuDNN 8.0构建的。在Anaconda中存在的版本是由Anaconda开发人员(或社区成员)使用CUDA 10.1和cuDNN 7.6.5构建的。此外,请注意,如果您自己构建TensorFlow GPU,则可以使用不同版本的CUDA和cuDNN。 - RoGuKa

0

更新于2022年4月26日,tensorflow 2.6.0版本可以胜任此工作, Python版本为3.8.13

回顾一下,在Anaconda Jupyter中遇到了相同的“GPU未显示”问题

'2.3.0'

遵循此链接中的过程,再次“GPU未显示”。

试错法:以下解决方法有效。

  1. 从 Navigator 中启动 CMD 窗口(相同环境);
  2. conda install tensorflow=2.6.0 –-channel conda-forge -y;
  3. 这需要一些时间(收集包、解析环境等);
  4. 在这种情况下,会自动更新 CUDA 和 cuDNN 版本到 11.3 和 8.2。
  5. 完成后,退出 Anaconda 并重新启动计算机。
  6. 回到 Jupyter,运行单元格:

cell

并检查结果:

2.6.0

[物理设备(name='/physical_device:CPU:0', device_type='CPU'), 物理设备(name='/physical_device:GPU:0', device_type='GPU')]


0

使用conda安装TensorFlow是管理TensorFlow本身以及CUDA和CUDNN的多个版本的更好方法。我最近创建了一个新的conda环境,并准备安装最新的TensorFlow。我也遇到了你提到的问题。我从conda install tensorflow-gpu的依赖列表中检查到和包都缺失了。由于Anaconda上tensorflow-gpu的最新版本是2.3,我认为这个问题已经被@GZ0在GitHub问题中回答过了。

下面是我的输出:

使用conda install tensorflow=2.3

PS > conda install tensorflow-gpu=2.3
## Package Plan ##

  environment location: C:\Anaconda3\envs\test_cuda_38

  added / updated specs:
    - tensorflow-gpu=2.3


The following packages will be downloaded:

    package                    |            build
    ---------------------------|-----------------
    absl-py-0.12.0             |   py38haa95532_0         176 KB
    aiohttp-3.7.4              |   py38h2bbff1b_1         513 KB
    astunparse-1.6.3           |             py_0          17 KB
    async-timeout-3.0.1        |   py38haa95532_0          14 KB
    blas-1.0                   |              mkl           6 KB
    blinker-1.4                |   py38haa95532_0          23 KB
    brotlipy-0.7.0             |py38h2bbff1b_1003         412 KB
    cachetools-4.2.1           |     pyhd3eb1b0_0          13 KB
    cffi-1.14.5                |   py38hcd4344a_0         224 KB
    chardet-3.0.4              |py38haa95532_1003         194 KB
    click-7.1.2                |     pyhd3eb1b0_0          64 KB
    coverage-5.5               |   py38h2bbff1b_2         272 KB
    cryptography-3.4.7         |   py38h71e12ea_0         643 KB
    cython-0.29.23             |   py38hd77b12b_0         1.7 MB
    gast-0.4.0                 |             py_0          15 KB
    google-auth-1.29.0         |     pyhd3eb1b0_0          76 KB
    google-auth-oauthlib-0.4.4 |     pyhd3eb1b0_0          18 KB
    google-pasta-0.2.0         |             py_0          46 KB
    grpcio-1.36.1              |   py38hc60d5dd_1         1.7 MB
    h5py-2.10.0                |   py38h5e291fa_0         841 KB
    hdf5-1.10.4                |       h7ebc959_0         7.9 MB
    icc_rt-2019.0.0            |       h0cc432a_1         6.0 MB
    idna-2.10                  |     pyhd3eb1b0_0          52 KB
    importlib-metadata-3.10.0  |   py38haa95532_0          34 KB
    intel-openmp-2021.2.0      |     haa95532_616         1.8 MB
    keras-applications-1.0.8   |             py_1          29 KB
    keras-preprocessing-1.1.2  |     pyhd3eb1b0_0          35 KB
    libprotobuf-3.14.0         |       h23ce68f_0         1.9 MB
    markdown-3.3.4             |   py38haa95532_0         144 KB
    mkl-2021.2.0               |     haa95532_296       115.5 MB
    mkl-service-2.3.0          |   py38h2bbff1b_1          49 KB
    mkl_fft-1.3.0              |   py38h277e83a_2         137 KB
    mkl_random-1.2.1           |   py38hf11a4ad_2         223 KB
    multidict-5.1.0            |   py38h2bbff1b_2          61 KB
    numpy-1.20.1               |   py38h34a8a5c_0          23 KB
    numpy-base-1.20.1          |   py38haf7ebc8_0         4.2 MB
    oauthlib-3.1.0             |             py_0          91 KB
    opt_einsum-3.1.0           |             py_0          54 KB
    protobuf-3.14.0            |   py38hd77b12b_1         242 KB
    pyasn1-0.4.8               |             py_0          57 KB
    pyasn1-modules-0.2.8       |             py_0          72 KB
    pycparser-2.20             |             py_2          94 KB
    pyjwt-1.7.1                |           py38_0          48 KB
    pyopenssl-20.0.1           |     pyhd3eb1b0_1          49 KB
    pyreadline-2.1             |           py38_1         145 KB
    pysocks-1.7.1              |   py38haa95532_0          31 KB
    requests-2.25.1            |     pyhd3eb1b0_0          52 KB
    requests-oauthlib-1.3.0    |             py_0          23 KB
    rsa-4.7.2                  |     pyhd3eb1b0_1          28 KB
    scipy-1.6.2                |   py38h66253e8_1        13.0 MB
    tensorboard-plugin-wit-1.6.0|             py_0         630 KB
    tensorflow-2.3.0           |mkl_py38h8557ec7_0           6 KB
    tensorflow-base-2.3.0      |eigen_py38h75a453f_0        49.5 MB
    tensorflow-estimator-2.3.0 |     pyheb71bc4_0         271 KB
    termcolor-1.1.0            |   py38haa95532_1           9 KB
    typing-extensions-3.7.4.3  |       hd3eb1b0_0          12 KB
    typing_extensions-3.7.4.3  |     pyh06a4308_0          28 KB
    urllib3-1.26.4             |     pyhd3eb1b0_0         105 KB
    werkzeug-1.0.1             |     pyhd3eb1b0_0         239 KB
    win_inet_pton-1.1.0        |   py38haa95532_0          35 KB
    wrapt-1.12.1               |   py38he774522_1          49 KB
    yarl-1.6.3                 |   py38h2bbff1b_0         153 KB
    ------------------------------------------------------------
                                           Total:       210.0 MB

使用 conda install tensorflow=2.1:
PS > conda install tensorflow-gpu=2.1
## Package Plan ##

  environment location: C:\Anaconda3\envs\test_cuda

  added / updated specs:
    - tensorflow-gpu=2.1


The following packages will be downloaded:

    package                    |            build
    ---------------------------|-----------------
    _tflow_select-2.1.0        |              gpu           3 KB
    absl-py-0.12.0             |   py37haa95532_0         175 KB
    aiohttp-3.7.4              |   py37h2bbff1b_1         507 KB
    astor-0.8.1                |   py37haa95532_0          47 KB
    async-timeout-3.0.1        |   py37haa95532_0          14 KB
    blas-1.0                   |              mkl           6 KB
    blinker-1.4                |   py37haa95532_0          23 KB
    brotlipy-0.7.0             |py37h2bbff1b_1003         337 KB
    cachetools-4.2.1           |     pyhd3eb1b0_0          13 KB
    cffi-1.14.5                |   py37hcd4344a_0         220 KB
    chardet-3.0.4              |py37haa95532_1003         192 KB
    click-7.1.2                |     pyhd3eb1b0_0          64 KB
    coverage-5.5               |   py37h2bbff1b_2         273 KB
    cryptography-3.4.7         |   py37h71e12ea_0         641 KB
    cudatoolkit-10.1.243       |       h74a9793_0       300.3 MB
    cudnn-7.6.5                |       cuda10.1_0       179.1 MB
    cython-0.29.23             |   py37hd77b12b_0         1.7 MB
    gast-0.2.2                 |           py37_0         155 KB
    google-auth-1.29.0         |     pyhd3eb1b0_0          76 KB
    google-auth-oauthlib-0.4.4 |     pyhd3eb1b0_0          18 KB
    google-pasta-0.2.0         |             py_0          46 KB
    grpcio-1.36.1              |   py37hc60d5dd_1         1.7 MB
    h5py-2.10.0                |   py37h5e291fa_0         808 KB
    hdf5-1.10.4                |       h7ebc959_0         7.9 MB
    icc_rt-2019.0.0            |       h0cc432a_1         6.0 MB
    idna-2.10                  |     pyhd3eb1b0_0          52 KB
    importlib-metadata-3.10.0  |   py37haa95532_0          34 KB
    intel-openmp-2021.2.0      |     haa95532_616         1.8 MB
    keras-applications-1.0.8   |             py_1          29 KB
    keras-preprocessing-1.1.2  |     pyhd3eb1b0_0          35 KB
    libprotobuf-3.14.0         |       h23ce68f_0         1.9 MB
    markdown-3.3.4             |   py37haa95532_0         144 KB
    mkl-2021.2.0               |     haa95532_296       115.5 MB
    mkl-service-2.3.0          |   py37h2bbff1b_1          48 KB
    mkl_fft-1.3.0              |   py37h277e83a_2         133 KB
    mkl_random-1.2.1           |   py37hf11a4ad_2         214 KB
    multidict-5.1.0            |   py37h2bbff1b_2          85 KB
    numpy-1.20.1               |   py37h34a8a5c_0          23 KB
    numpy-base-1.20.1          |   py37haf7ebc8_0         4.1 MB
    oauthlib-3.1.0             |             py_0          91 KB
    opt_einsum-3.1.0           |             py_0          54 KB
    protobuf-3.14.0            |   py37hd77b12b_1         240 KB
    pyasn1-0.4.8               |             py_0          57 KB
    pyasn1-modules-0.2.8       |             py_0          72 KB
    pycparser-2.20             |             py_2          94 KB
    pyjwt-1.7.1                |           py37_0          49 KB
    pyopenssl-20.0.1           |     pyhd3eb1b0_1          49 KB
    pyreadline-2.1             |           py37_1         143 KB
    pysocks-1.7.1              |           py37_1          28 KB
    requests-2.25.1            |     pyhd3eb1b0_0          52 KB
    requests-oauthlib-1.3.0    |             py_0          23 KB
    rsa-4.7.2                  |     pyhd3eb1b0_1          28 KB
    scipy-1.6.2                |   py37h66253e8_1        12.8 MB
    six-1.15.0                 |   py37haa95532_0          51 KB
    tensorboard-plugin-wit-1.6.0|             py_0         630 KB
    tensorflow-2.1.0           |gpu_py37h7db9008_0           4 KB
    tensorflow-base-2.1.0      |gpu_py37h55f5790_0       105.3 MB
    tensorflow-estimator-2.1.0 |     pyhd54b08b_0         251 KB
    tensorflow-gpu-2.1.0       |       h0d30ee6_0           3 KB
    termcolor-1.1.0            |   py37haa95532_1           9 KB
    typing-extensions-3.7.4.3  |       hd3eb1b0_0          12 KB
    typing_extensions-3.7.4.3  |     pyh06a4308_0          28 KB
    urllib3-1.26.4             |     pyhd3eb1b0_0         105 KB
    werkzeug-0.16.1            |             py_0         258 KB
    win_inet_pton-1.1.0        |   py37haa95532_0          35 KB
    wrapt-1.12.1               |   py37he774522_1          49 KB
    yarl-1.6.3                 |   py37h2bbff1b_0         151 KB
    ------------------------------------------------------------
                                           Total:       745.0 MB

因此,您可以按照@GZ0和@geometrikal的建议,在Windows平台上从Anaconda安装tensorflow-gpu的最新版本(v2.3),或者只需使用 conda install tensorflow-gpu=2.1 来获取最新和正确的环境。
请注意,tensorflow-gpu v2.1仅支持Python 3.5-3.7之间的版本。

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接