当地的GPU

默认构建的TensorFlow将使用NVIDIA®GPU,如果它是可用的,并安装了适当的驱动程序,否则只使用CPU。

在每个平台上使用GPU版本TensorFlow的先决条件如下所述。请注意,在所有平台(macOS除外)上,您必须运行具有CUDA®Compute Capability 3.5或更高版本的NVIDIA®GPU。请参阅支持cuda的GPU卡

要使TensorFlow使用本地NVIDIA®GPU,您可以安装以下组件:

要在Ubuntu 20.04上安装所需的NVIDIA组件,您可以在终端运行以下命令:

wgethttps://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pinsudomv cuda-ubuntu2004。销/etc/apt/preferences.d / cuda -库销- 600sudoapt-key副词——fetch-keyshttps://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pubsudoadd-apt-repository"deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"sudoapt-get更新sudoapt-get可能是安装cuda-11-2 libcudnn8=8.1.1.33-1+cuda11.2sudoApt-mark hold libcudn8

首先,确认您有一个合适的GPU。本文介绍如何检测您的显卡是否使用NVIDIA®GPU:

http://nvidia.custhelp.com/app/answers/detail/a_id/2040/~/identifying-the-graphics-card-model-and-device-id-in-a-pc

一旦您确认您拥有NVIDIA®GPU,以下文章将介绍如何安装所需的软件组件,包括CUDA Toolkit v11.2和cuDNN 8.1:

https://www.tensorflow.org/install/gpu#hardware_requirements

请注意,最后一个组件(cuDNN)的安装文档有点稀疏。一旦您加入NVIDIA®开发人员程序并下载包含cuDNN的zip文件,您需要提取zip文件并将提取位置添加到您的系统PATH。

安装所需软件的细节因Linux版本而异,因此请仔细查看NVIDIA®文档,以确保正确安装所有软件。

TensorFlow不支持在mac上使用卡塔尔世界杯欧洲预选赛赛程表NVIDIA gpu。然而,对于ARM (M1, M2) mac,苹果提供了一个可以利用GPU的自定义构建的TensorFlow。install_tensorflow ()将检测它是否在Arm Mac上运行,并自动安装相应的Python包(tensorflow-macostensorflow-metal).

Conda是Arm mac上唯一支持卡塔尔世界杯欧洲预选赛赛程表的安装方法。此外,不支持运行在Rosetta下的x86_64版本的R;卡塔尔世界杯欧洲预选赛赛程表需要Arm64版本的R。

如果你有一个Arm版本的R,你可以这样测试:

R.version

如果返回“aarch64”,您正在使用正确的r版本,如果它返回“x86_64”,那么你需要安装一个不同版本的R。你可以通过导航到https://cloud.r-project.org/bin/macosx/并选择合适的软件包(例如,“R-4.2.1-arm64.pkg”)或使用如下的安装管理器钻井平台和运行钻机添加释放

你可以像这样配置reticulate来使用conda Python:

网状::install_miniconda()tensorflow::install_tensorflow()

请注意,install_miniconda ()自动创建一个名为r-reticulate,这就成为网纹使用的默认环境。

测试

你可以测试TensorFlow是否可以使用本地GPU运行以下程序:

图书馆(tensorflow)特遣部队配置list_physical_devices“图形”

如果你看到这样一个列表:

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

那么这意味着TensorFlow配置正确。如果你看到一个空的列表(),则TensorFlow不使用GPU。

为了支持卡塔尔世界杯欧洲预选赛赛程表安装,请打开aGithub的问题