前言:
在此之前,写了一篇基于GTX1660显卡的CUDA 11.2的环境搭建,参考链接: 基于tensorflow-gpu + E5 CPU + GTX1660 SUPER + TESLA-P40 显卡 安装CUDA(11.8.0)
本文为基于NVIDIA GPU (TESLA P40) 进行 tensorflow(CUDA) 计算的环境配置说明,如果电脑中安装有cuDNN 且Tensorflow支持的话,会默认优先使用GPU进行计算,如果有多块GPU ,需要代码指定使用哪一块显卡进行计算,如果配置CUDA失败,会使用CPU计算
| # | 软件 | 说明 | 查询地址 |
|---|---|---|---|
| 1 | 操作系统 | WINDOWS10 | - |
| 2 | cuda显卡 | 在cuda-gpus列表查询当前显卡是否支持cuda功能及Compute Capability版本 | cuda-gpus |
| 3 | CUDA 版本 | 通过查询Tensorflow-GPU版本列表,当前文档中最高支持CUDA 11.2, 但是实际上使用CUDA(11.8.0)版本配合对应版本cuDNN依然可以使用低版本的Tensorflow-GPU | Tensorflow-GPU |
| 4 | 显卡驱动 | 找到与CUDA 版本对应的CUDA开发工具的-显卡驱动即可 | cuda-toolkit-release |
| 5 | cuDNN | 在cudnn-archive版本列表中显示 for CUDA 11.x 或者指定版本即可 | cudnn |
注:
1. Compute Capability版本对应的支持功能,可以查询cuda-function进行查询
2. cudnn下载需要注册NVIDIA的开发者账号
硬件、软件配置
1.文档涉及硬件配置
| # | 设备 | 产品 |
|---|---|---|
| 1 | 主机 | DELL R730XD |
| 1 | CPU | E5-2686V4 * 2 |
| 2 | 内存 |


165

被折叠的 条评论
为什么被折叠?



