一、CUDA安装
检查电脑是否支持GPU类型。
点击设备管理器---->显示示配器
进入官网查看显卡是否支持CUDA:https://developer.nvidia.com/cuda-gpus
安装CUDA
请看清楚您要安装的tensorflow版本对应的CUDA版本号!
请看清楚您要安装的tensorflow版本对应的CUDA版本号!
点击链接查看:https://tensorflow.google.cn/install/source#linux
在https://developer.nvidia.com/cuda-toolkit-archive
链接中下载
由于官网下载速度慢,因此找到网页下载按钮。鼠标右击,复制链接地址。在迅雷中打开链接进行下载。
取消勾选GeForce Experience
如果电脑上本身就有Visual Studio Integration,要将这个取消勾选,避免冲突
点开Driver comonents,Display Driver这一行,前面显示的是Cuda本身包含的驱动版本是411.31
如果你电脑目前安装的驱动版本号新于Cuda本身自带的驱动版本号,那一定要把这个勾去掉。否则会安装失败(相同或者小于驱动版本号的话,就不用去取勾了)
剩下的直接安装即可。
安装成功后 在 命令行输入 nvcc -V 会显示版本信息 则安装成功。
如果没有 查看系统环境变量 的 path 中是否有CUDA 的相关目录
安装CUDNN
请看清楚您要安装的tensorflow版本对应的CUDNN版本号!
CUDNN官网链接:https://developer.nvidia.com/rdp/cudnn-download
选择相应的版本号:
解压CUDNN,将其复制到CUDA文件夹下
系统环境配置
添加环境路径:
控制面板——>系统和安全——>系统——>高级系统设置
点击环境变量,在系统变量中找到PATH
双击打开查看,添加好以下路径:
安装tensorflow
pip install tensorflow-gpu==1.13.1
安装完成!
参考博文:https://www.cnblogs.com/Assist/p/11002756.html
https://www.sohu.com/a/323645104_100058348