Window10 NVIDIA显卡 Tensorflow 2.1 GPU 安装和测试

原创 专栏收录该内容
3 篇文章 0 订阅

Window10 NVIDIA显卡 Tensorflow 2.1 GPU 安装和测试

1. 硬件要求

支持以下带有 GPU 的设备:
CUDA® 计算能力为 3.5 或更高的 NVIDIA® GPU 卡。请参阅支持 CUDA 的 GPU 卡列表。

2. 安装NVIDIA GPU驱动,需要根据你的显卡下载驱动

下载地址:
https://www.nvidia.com/download/index.aspx?lang=en-us

运行这个命令看nvidia显卡的驱动的版本号 ----nvidia-smi
在这里插入图片描述
或在自己桌面右键打开英伟达控制面板,查看显卡信息。

3、查看本机的CUDA驱动适配版本

桌面右键打开英伟达控制面板,点击帮助->系统信息->组件
在这里插入图片描述
可以看到本机支持的是CUDA 10.2 版本,表示是不支持更高版本的。如果你升级了驱动,可能会支持更高版本,也可能不会提升。

所以就必须安装 11.2 及以下的版本。

4. 下载CUDA

CUDA的下载地址为:https://developer.nvidia.com/cuda-downloads
找到自己显卡的版本号。进行下载和安装。
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
按"Windows+R"组合键,输入“cmd”进入,并查看CUDA 的版本信息。
在这里插入图片描述
5. 下载cuDNN

cuDNN下载页面:https://developer.nvidia.com/rdp/cudnn-download

下载cuDNN是需要登录英伟达开发者账户的,注册一个并填写问卷就行了,很简单。
注意:必须选择和你安装的CUDA匹配的版本。
在这里插入图片描述
在这里插入图片描述
6、安装CUDA和cuDNN

找到你下载的CUDA,无脑安装就行了。当然如果你想自定义的话要记住你选择的安装路径。

CUDA安装完成后,打开powershell,执行nvcc -V ,成功的话会返回cuda版本号。

解压cuDNN压缩包,可以看到bin、include、lib目录
在这里插入图片描述
打开 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

找到你安装的版本目录,打开,找到bin、include、lib目录,将cuDNN压缩包内对应的文件复制到bin、include、lib目录。

注意:是复制文件到bin、include、lib目录,不是复制目录。

7、添加环境变量

你需要在系统环境变量的Path项下添加几个路径
在桌面上找到“计算机”右键“属性”;在打开的窗口内找到“高级系统设置”;打开进入“环境变量”。在“系统变量”找到路径“path",点击 编辑 – > 新建、浏览
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
需要添加下面两个路径,这就是说为什么要记住你的安装路径了,我使用的是默认的安装路径。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\lib\x64
注意:选择你安装的路径

8、检查安装结果

打开 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\demo_suite
在此路径下打开bandwidthTest进行查看。

  • 0
    点赞
  • 0
    评论
  • 1
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值