Windows10搭建Tensorflow2 gpu环境(2021.11.15)

1.Anaconda创建tensorflow gpu环境

conda create -n tf2-gpu1 python=3.7

2.查看NVIDIA驱动版本

右键->NVIDIA控制器

image

image

我电脑最新刚买的,可以看到对应的NVIDIA驱动版本是11.4。

3.准备各版本

CUDA Toolkit 11.4.0、cuDNN v8.2.x

tensorflow-gpu:2.7.0(最新版)、Anaconda:最新版

注意:下载CUDA Toolkit 11.4.0即可,不能下载高于此版本的。也不能按照网上教程下载CUDA10.x版本的。否则会出现类似如下问题:cudart64_101.dll not found(后面会详细说)

image

4.安装CUDA和cuDNN

(1)安装CUDA

CUDA下载地址: https://developer.nvidia.com/cuda-toolkit-archive

image

安装CUDA过程:

image

image

Visual Studio Integration不要选,怕和本地环境冲突。

image

这个也不要选,本地已经有这个,就不用安装了。

接下来等安装完成即可。安装完毕后,按Win+R打开Windows并输入cmd,随后输入nvcc -V

image

(2)下载cuDNN神经网络加速库

cuDNN下载地址:https://developer.nvidia.com/rdp/cudnn-archive

image

下载完毕后,将cuda文件复制到CUDA的安装目录下,如图所示

image

(3)配置环境变量

将cuDNN神经网络加速库添加到环境变量中,如图所示

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\cuda\bin

image

其他几个没有,也加上,一般CUDA安装好了会默认加上,不用手动加

5.安装Tensorflow

在tenforflow-gpu环境下输入:pip install -U tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

image

测试tensorflow-gpu是否安装成功:

import tensorflow as tf
tf.__version__
tf.test.is_gpu_available()

image

如图所示,即证明tensorflow-gpu安装成功。

在pycharm配置环境

image

问题记录:

问题一:找不到cudart的dll文件

在cmd中就出现如下问题:cudart64_101.dll not found。说明版本不对,仔细检查版本。我之前安装了CUDA10.2,出现了这个问题。然后我删除了这个版本,查看本地NVIDIA驱动版本是11.4,于是下载对应的11版本。

问题二:在cmd测试正常,但是在pycharm中出现了:cudart64_101.dll not found。这是环境变量的问题,将系统配好的环境变量加入到IDE的环境变量中,让IDE能找到CUDA的安装目录即可。

在环境变量没问题的情况下,我的解决方案:关掉pycharm,然后重启,重新测试即可

posted @ 2021-11-15 18:31  Vinta  阅读(225)  评论(0编辑  收藏  举报