anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5

背景:我的是台式机,显卡3070,准备安装tf - gpu ,且准备在pycharm和jupyter上工作

  1. 新建虚拟环境
    conda create -n tf python=3.8
  2. 激活虚拟环境
conda activate tf 

安装cuda11.6和cudnn8.5

anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第1张图片
下载地址:https://developer.nvidia.com/cuda-toolkit-archive
anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第2张图片
anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第3张图片
正常安装就好,地址可以改,
anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第4张图片
anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第5张图片
anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第6张图片
anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第7张图片

看清楚必须要求当前版本>=新版本,如大于,则取消勾选,如等于也可以随便选择,如果小于,则必须勾选,也就是说,我们要安装最新的版本,切记!!!!
anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第8张图片
上面地址 不要变!!! 不要变 然后此处记得截图,后面配置环境变量可能会用到
然后剩下的一路保持默认安装就好。
anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第9张图片
可以看到这里系统已经自动帮我们配置好了,如果这里没有,则按照上面的截图的地址来配置

然后打开cmd 输入nvcc -V 看是否安装成功

在这里插入图片描述

cuDNN神经网络加速库安装

顾名思义:这是针对的是神经网络进行加速的,https://developer.nvidia.com/rdp/cudnn-archive,进入官网,cudnn官网,
anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第10张图片
anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第11张图片
此处,需要登录NVIDIA的账户。登录完成后才能下载,大家可以注册一下,只要邮箱,很快,预计1分钟

然后解压
anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第12张图片
然后打开下图的路径,里面是cuda的源文件,然后无脑吧上图的4个文件全部复制到下图的路径中,可以看到,下图后面四个箭头就是我复制好的,如果重名,直接替换,cuDNN其实就是CUDA的一个补丁而已
anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第13张图片
配置环境变量

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\include

看清楚哈,是Path里面,新建,把这四个加进去,保存
anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第14张图片
配置好环境后,我们需要验证环境变量是否配置成功,打开cmd,输入下面指令

cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\extras\demo_suite

然后分别输入下面两个命令

.\bandwidthTest.exe

anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第15张图片

.\deviceQuery.exe

anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第16张图片
两个都是PASS,则说明成功!
cmd 输入 nvidia-smi
anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第17张图片
TensorFlow的gpu版本安装

我试了很多博主的,我用这个命令 pip install -U tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple 发现安装的是tf2.10的版本,我想要2.9 的,(“-U”参数指定如果已安装此包,则进行升级命令。)
最后一种方法成功,

pip install  tensorflow-gpu==2.9.0  

我是在cmd窗口,进入虚拟环境,然后输入上面命令的
(打开cmd 输入conda activate tf 即可进入虚拟环境 )

然后 进入python环境,若无错误信息,输入tf.test.is_gpu_available()

anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第18张图片
或者输入 tf.config.list_physical_devices(‘GPU’)
在这里插入图片描述
查看可用的gpu数量:print(‘Num GPUs Available:’,len(tf.config.list_physical_devices(‘GPU’)))
在这里插入图片描述
查看TensorFlow的版本信息 tf.version

在这里插入图片描述
记录一下,这是总图 2022-11-7 16:38写的
anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第19张图片

安装jupyter内核

然后,还有个问题,在虚拟环境里,进入jupyter,无内核
anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第20张图片
直接打开命令窗,进入虚拟环境,直接键入:

conda install nb_conda

关闭命令窗,再次打开,键入jupyter notebook,进入后,发现有了!!
大功告成!!!记录一次安装ttensorflow的经历,花费了我半天时间!!喜欢的话点个赞!未经允许,禁止转载!!!

anaconda虚拟环境安装tensorflow GPU 2.9.0 cuda11.6 cuDNN 8.5_第21张图片

本文写于2022-11-7 16:38

你可能感兴趣的:(tensorflow,tensorflow,python,深度学习)