Win10 Anaconda安装tensorflow-gpu==2.5.0

最新推荐文章于 2024-07-11 13:32:44 发布

原创

最新推荐文章于 2024-07-11 13:32:44 发布 · 1.2k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #深度学习

本文详细介绍了在Windows 10上安装CUDA 11.4、cuDNN以及Anaconda中的TensorFlow-GPU 2.5.0的步骤。首先，通过NVIDIA官网下载CUDA并进行自定义安装，注意避开可能导致失败的选项。接着，下载并解压cuDNN文件，将其内容复制到CUDA安装目录。然后，在Anaconda中创建新环境并安装指定版本的TensorFlow-GPU。最后，针对可能出现的找不到cudart64_110.dll等错误，将这些文件复制到系统目录下以解决问题。

1.安装CUDA11.4

下载CUDA11.4.1，地址链接https://developer.nvidia.com/cuda-toolkit-archive
在这里插入图片描述

下载后选择默认安装位置。
自定义安装。
安装过程注意事项：
（1）若没有安装VS，则取消勾选
（2）Driver选项里注意版本，若系统版本高于CUDA的驱动版本，则取消
（3）其余事项可百度
安装失败的原因可能是安装过程勾选的有误。

2.安装cudnn

下载对应版本的cudnn，下载地址https://developer.nvidia.com/rdp/cudnn-archive
在这里插入图片描述
下载完毕后解压压缩包，解压后文件名为cuda，将其改为cudnn，然后将bin、include、lib/x64中的文件复制到CUDA/v11.4/对应的文件夹中。最后将cudnn整个文件夹复制到CUDA/v11.4/中。