Jupyter Notebook 在导入 Tensorflow 时未检测到 GPU

Question

在 jupyter 笔记本中，我正在为我的项目导入张量流。 如果训练确实在 GPU 上工作那就更好了，所以我尝试运行 !pip3 install --upgrade tensorflow 命令，但我收到如下错误： 子进程退出并出现错误

python setup.py Egg_info 未成功运行。 退出代码：1

[44行输出] 回溯（最近一次调用最后一次）： 文件“C:\Users\ASUS naconda3\Lib\site-packages\setuptools_vendor\packaging Equirements.py”，第 35 行，在 init 中 解析 = _parse_requirement(requirement_string) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ 文件“C:\Users\ASUS naconda3\Lib\site-packages\setuptools_vendor\packaging_parser.py”，第 64 行，在 parse_requirement 中 返回_parse_requirement（Tokenizer（源，规则= DEFAULT_RULES）） ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ ^^^^^^^^ 文件“C:\Users\ASUS naconda3\Lib\site-packages\setuptools_vendor\packaging_parser.py”，第 82 行，在 _parse_requirement 中 url、说明符、标记 = _parse_requirement_details(tokenizer) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ 文件“C:\Users\ASUS naconda3\Lib\site-packages\setuptools_vendor\packaging_parser.py”，第 126 行，在 _parse_requirement_details 中 标记=_parse_requirement_marker( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^ 文件“C:\Users\ASUS naconda3\Lib\site-packages\setuptools_vendor\packaging_parser.py”，第 147 行，在 _parse_requirement_marker 中 tokenizer.raise_syntax_error( 文件“C:\Users\ASUS naconda3\Lib\site-packages\setuptools_vendor\packaging_tokenizer.py”，第 165 行，在 raise_syntax_error 中 引发解析器语法错误（ setuptools.extern.packaging._tokenizer.ParserSyntaxError：预期结束或分号（在名称之后并且没有有效的版本说明符） python_version>“3.7” ^

上述异常是导致以下异常的直接原因：

回溯（最近一次调用最后一次）： 文件“”，第 2 行，位于 文件“”，第 34 行，位于 文件“C:\Users\ASUS\AppData\Local\Temp\pip-install-1zpm1ck6 ensorflow-gpu_76e71b3a0add4c93b61fcedc949da126\setup.py”，第 40 行，位于 setuptools.setup() 文件“C:\Users\ASUS naconda3\Lib\site-packages\setuptools_init_.py”，第 102 行，在安装程序中 install_setup_requires（属性） 文件“C:\Users\ASUS naconda3\Lib\site-packages\setuptools_init.py”，第 73 行，位于 _install_setup_requires 中 dist.parse_config_files(ignore_option_errors=True) 文件“C:\Users\ASUS naconda3\Lib\site-packages\setuptools\dist.py”，第 655 行，在 parse_config_files 中 self._finalize_requires() 文件“C:\Users\ASUS naconda3\Lib\site-packages\setuptools\dist.py”，第 390 行，位于 _finalize_requires 中 self._normalize_requires() 文件“C:\Users\ASUS naconda3\Lib\site-packages\setuptools\dist.py”，第 405 行，位于 _normalize_requires 中 self.install_requires = 列表(map(str, _reqs.parse(install_requires))) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ 文件“C:\Users\ASUS naconda3\Lib\site-packages\setuptools_vendor\packaging Equirements.py”，第 37 行，在 init 中 从 e 引发 InvalidRequirement(str(e)) setuptools.extern.packaging.requirements.InvalidRequirement：预期结束或分号（在名称之后并且没有有效的版本说明符） python_version>“3.7” ^ [输出结束]

注意：此错误源自子进程，并且可能不是 pip 的问题。 错误：元数据生成失败

生成包元数据时遇到错误。

参见上面的输出。

注意：这是上面提到的包的问题，而不是 pip 的问题。 提示：详细信息请参见上文。 请提出修复建议

我尝试在 Jupyter 笔记本中使用 GPU 来完成我的项目。 我期望 tf.test.is_gpu_available() 给出 true 作为输出，但得到 false

Answer 1

你的Python版本是什么？如果是

3.7

或更低，那么您需要将其升级到

3.9

和

3.11

之间的较新版本。我建议使用 Anaconda 创建一个新的虚拟环境：

conda create -n <env_name> python=3.9

此外，您的操作系统似乎是Windows，并且上述Tensorflow版本

2.10

不提供Windows上的GPU支持。升级 python 版本后，请从终端使用以下命令：

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
# Anything above 2.10 is not supported on the GPU on Windows Native
python -m pip install "tensorflow<2.11"
# Verify the installation:
python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

如果上述步骤成功运行，您应该看到：

[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]

在您的终端中。

确保您已更新 GPU 驱动程序并为您的 GPU 安装 CUDA。更多详情请参阅这里。