ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

安装TensorFlow GPU版本

2021-08-03 20:32:33  阅读:209  来源: 互联网

标签:版本 GPU 驱动 CUDA install NVIDIA TensorFlow 安装


首先需要准备的东西:Anaconda环境

查看自己GPU型号并安装英伟达驱动程序

首先要知道自己电脑的GPU版本型号:右击我的电脑->管理->设备管理器。

比如我的时NVIDIA GeForce GTX 1060 6GB,

英伟达GeForce系列 ,型号为GTX1060 6GB运行内存,知道这个以后再去官网去看自自己相应的软件驱动

查询自己的GPU所对应的驱动:CUDA GPU | NVIDIA Developer 查询到我的算力时6.1

下载GPU驱动   Download Drivers | NVIDIA

点击search进入下载驱动程序

建议就按照软件的默认路径装,这样可能会把上一把不正确的删除掉,安装完成后的界面如下:

安装cudatoolkit  

CUDA与显卡驱动:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

TensorFlow-GPU与CUDA cudnn Python版本关系:https://tensorflow.google.cn/install/source_windows?hl=en#gpu
 

打开这个   输入命令:

  1. conda install cudatoolkit=10.1
  2. conda install cudnn=7.6
  3. pip install tensorflow-gpu==2.1.2

标签:版本,GPU,驱动,CUDA,install,NVIDIA,TensorFlow,安装
来源: https://blog.csdn.net/m0_37755481/article/details/119353326

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有