1.安装cuda
表明我的电脑中没有cuda
2.我应该安装什么版本的cuda呢?
查找到驱动版本的程序信息,我的显卡驱动版本是512.13.
去英伟达官网查找能安装的cuda版本
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
查找到自己能安装的cuda版本为11.6多。
3.怎样安装cuda?
cuda官网下载,https://developer.nvidia.com/cuda-toolkit-archive
自定义安装
没有使用visual Studio,所以不能勾选
然后安装要等待一段时间
显示环境变量已经配置好了
检查cuda是否安装好
查看cuda设置的环境变量
找到cuda的安装目录,找到“nvcc.exe”
4.cuDNN神经网络加速库安装
cuDNN Archive | NVIDIA Developerhttps://developer.nvidia.com/rdp/cudnn-archive
把这四个文件复制到该文件夹下,只需要强制粘贴
cuDNN只是cuda的一个补丁,优化深度学习用的。
配置环境变量,只需要把粘贴好的四个文件所在路径创建环境变量。
5.检查cuDNN是否安装好?
怎样在命令行打开指定文件夹?
因为这个C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\extras\demo_suite路径好长,直接找到文件所在位置,在路径栏里输入cmd,就可以在cmd中打开这个文件夹。
两个pass表明cuDNN配置成功。
6.tensorflow的GPU安装
命令行输入pip install -U tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple
出现这个提示:[notice] To update, run: python.exe -m pip install --upgrade pip
下面警告好像是说我C盘里有python.exe,我也去运行下。
已经安装过了,没报错我就不改了。
检查tensorflow-gpu是否安装成功?
在python环境下输入如下语句,如果下面显示True就表示安装成功。
查看服务器类型
tensorflow的版本信息:
到这里TensorFlow就安装成功了。