tensorflow-gpu安装

欲哭无泪

基本环境要求

本人电脑显卡:NVIDIA GeForce GTX 1050
CUDA版本:CUDA 10.0
cuDNN版本:cuDNN v7.6.5

安装CUDA

  1. 官网网址:https://developer.nvidia.com/cuda-10.0-download-archive
  2. 选择Windows 平台,x86_64 架构,10 系统,exe(local)



    3.安装

  • 安装程序选择列表,在这里选择需要安装和取消不需要安装的程序。在 CUDA 节点下,取消”Visual Studio Integration”一项;
  • 在“Driver components”节点下,比对目前计算机已经安装的显卡驱动“Display Driver”的版本号“Current Version”和 CUDA 自带的显卡驱动版本号“New Version”,如果“Current Version”大于“New Version”,需要取消“Display Driver”的勾,如果小于或等于,则默认勾选即可。
    4.测试是否安装成功
    cmd命令行输入"nvcc -V",若显示 release 10.0,V10.0.130说明成功了


安装CuDNN

1.官网网址:https://developer.nvidia.com/cudnn,要先注册

2.由于我们安装的是cuda1.0,所以我们切记要安装对应的版本!!!


该地址链接为:https://developer.nvidia.com/rdp/cudnn-archive#a-collapse764-10
3.下载完成 cuDNN 文件后,解压并进入文件夹,我们将名为“cuda”的文件夹重命名为
“cudnn765”,并复制此文件夹。
4.进入 CUDA 的安装路径 C:\Program Files\NVIDIA GPU
Computing Toolkit\CUDA\v10.0,粘贴“cudnn765”文件夹即可
5.环境变量 Path 配置 上述 cudnn 文件夹的复制即已完成 cuDNN 的安装,但为了让系统能够感知到 cuDNN 文件的位置,我们需要额外配置 Path 环境变量。打开文件浏览器,在“我的电脑”上右击,选择“属性”,选择“高级系统属性”,选择“环境变量”,在“系统变量”一栏中选中“Path”环境变量,选择“编辑”,如图 1.33 所示。选择“新建”,输入我们 cuDNN 的安装路径“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\cudnn765\bin”,并通过“向上移动”按钮将这一项上移置顶。

TensorFLow 安装

cmd中...
# 若之前下载过tensorflow-cpu,请先卸载掉
conda uninstall tensorflow
# 下载tensorflow-gpu
pip install tensorflow-gpu==2.0.0 -i https://pypi.douban.com/simple
# 进入python
python
# 导入包
import tensorflow as tf
# 测试是否可以成功使用tensorflow_gpu
tf.test.is_gpu_available()
# 若显示True,则代表安装成功!
# 查看版本号
tf.__version__
# 2.0.0
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容