windows10安装cuda11.4

cuda11.4安装

前言

tensorflow1.12之后gpu使用cuda10.0对应的驱动,不要求安装cuda,[但pytorch要求安装cuda和cudnn(暂不确定)],如果本机的驱动版本小于cuda10.0对应的驱动版本,建议安装cuda10.x的驱动来覆盖本机的驱动,不用卸载再安装驱动。CUDA、cuDNN和tensorflow-gpu之间是需要对应版本的。

TensorFlow pip软件包现在使用CUDA11和cuDNN 8.0.2构建。目前最新版本TensorFlow 2.4.0

https://github.com/tensorflow/tensorflow/releases

官网tensorlow对应GPU支持情况查询:

https://tensorflow.google.cn/install/source_windows#gpu

目前最高支持:

版本Python 版本编译器构建工具cuDNNCUDA
tensorflow_gpu-2.4.03.6-3.8MSVC 2019Bazel 3.1.08.011.0

下载cuda11.4和cudnn8.2

cuda

cudnn

不详讲了,本机下载对应的文件:
cuda_11.4.1_471.41_win10.exe

cudnn-11.4-windows-x64-v8.2.2.26.zip

查看本机驱动版本

打开nvidia的控制面板 -> 系统信息

驱动是cuda11.4版本,安装cuda11.4时,如果你的低于此版本则需要覆盖驱动 

安装过程

  1. 选择自定义安装,可以选择安装驱动,覆盖本机的驱动
    在这里插入图片描述
  2. 不要选Visual Studio Integration,即使选了也不能成功安装
    在这里插入图片描述
  3. 如果本机的驱动版本(当前版本)小于cuda对应的版本(新版本),则选择,否则不选。如果当前版本小于新版本,并且不覆盖安装,之后电脑会频繁蓝屏或死机
    在这里插入图片描述
  4. 记住安装位置,tensorflow要求配置环境
    在这里插入图片描述
  5. 安装完成后配置环境,Path需要手动添加如下路径,对应上一步的安装路径

 

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\extras\CUPTI\lib64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.4\bin\win64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.4\common\lib\x64

安装cudnn

将下载的cudnn文件解压,复制里面的文件夹到cuda的安装目录即可
bin
include
lib

验证

打开命令窗口cmd,输入验证命令 nvcc -V
C:\Users\Administrator\Desktop> nvcc -V
nvcc: NVIDIA ® Cuda compiler driver
Copyright © 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:32:27_Pacific_Daylight_Time_2019
Cuda compilation tools, release 10.2, V10.2.89

输出如上信息表示 安装成功!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值