记一次在Windows中安装CUDA及tensorflow-gpu

最新推荐文章于 2023-03-30 22:43:47 发布

lonely虾

最新推荐文章于 2023-03-30 22:43:47 发布

阅读量146

点赞数

本文链接：https://blog.csdn.net/m1233321/article/details/108808166

版权

本文详细记录了在Windows操作系统中安装CUDA 10.1和cuDNN v7.6.0的步骤，包括安装CUDA的注意事项、cuDNN的下载配置，以及如何在Anaconda环境中部署并测试TensorFlow-GPU的安装成功。

摘要由CSDN通过智能技术生成

1、安装CUDA（10.1）

https://developer.nvidia.com/cuda-10.1-download-archive-base
注意：最好安装到默认地址。
补充：安装完成后用nvcc -V检查是否安装成功，以下为安装正确。

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Fri_Feb__8_19:08:26_Pacific_Standard_Time_2019
Cuda compilation tools, release 10.1, V10.1.105

2、下载cuDNN( cuDNN v7.6.0 (May 20, 2019), for CUDA 10.1)

https://developer.nvidia.com/cudnn

配置

复制 cuda\bin\cudnn64_7.dll 到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin\
复制 cuda\include\cudnn.h    到  C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include\
复制 cuda\lib\x64\cu