经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置:技术经验 » 大数据云AI » 人工智能基础 » 查看文章
安装cuda9.0+cudnn v7+python3.5.3+tensorflow
来源:cnblogs  作者:LeaveMeAlone1995  时间:2018/11/3 9:44:31  对本文有异议

 

本机设备 windows10 gtx1060

安装软件及下载地址

python-3.5.3-amd64  链接:https://pan.baidu.com/s/1I3oIDatMgvDLEtaPtvuWSw  提取码:a7dp

cuda_9.0.176_win10  链接:https://pan.baidu.com/s/1Iao-jCwlQhLIq5D28lyWmQ 提取码:f9ch

cudnn-9.0-windows10-x64-v7.3.1.20 链接:https://pan.baidu.com/s/14RSIsbxtqfJe0tSIRQAznA 提取码:8j7g

首先安装python,直接运行安装包,安装完成后输入

  1. python -V

  

安装cuda_9.0.176_win10

因为gtx1060比cuda9.0要新,所以会提示下图内容,不过没有任何影响

接受许可协议后,选择自定义安装

 

我的电脑本身已经带有physX和显卡驱动程序了,所以取消勾选

值得注意的是Visual Studio Integration不需要勾选,这个应该是使用c或c++直接操作cuda需要用到的,如果只在python里面使用tensorflow的gpu版本可以不勾选。

等待安装完成后,进入环境变量设置(可以直接用小娜搜索,按windows键输入“环境”,基本上就可以找到)

 

安装完后成会出现以上两条CUDA路径

需要再次基础上加上

  1. CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0
  2. CUDA_LIB_PATH = %CUDA_PATH%\lib\x64
  3. CUDA_BIN_PATH = %CUDA_PATH%\bin
  4. CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64
  5. CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64

  

安装cudnn

解压cudnn到合适的位置后

在环境变量里PATH中添加对应目录下的bin目录即可

保存环境变量后,开启cmd命令行

输入set CUDA进行验证,环境变量是否设置正确

输入nvcc -V 验证cuda是否安装正确

 

 

安装tensorflow-gpu版本

在cmd中升级pip

  1. python -m pip install --upgrade pip

安装tensorflow-gpu

  1. pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu

验证方案

运行cmd指令,输入

  1. python
  2. import tensorflow as tf
  3. tf.__version__ # 显示tensorflow版本 ,如图为1.11.0
  4. sess = tf.Session() # 显示显卡型号

 

 友情链接:直通硅谷  点职佳  北美留学生论坛

本站QQ群:前端 618073944 | Java 606181507 | Python 626812652 | C/C++ 612253063 | 微信 634508462 | 苹果 692586424 | C#/.net 182808419 | PHP 305140648 | 运维 608723728

W3xue 的所有内容仅供测试,对任何法律问题及风险不承担任何责任。通过使用本站内容随之而来的风险与本站无关。
关于我们  |  意见建议  |  捐助我们  |  报错有奖  |  广告合作、友情链接(目前9元/月)请联系QQ:27243702 沸活量
皖ICP备17017327号-2 皖公网安备34020702000426号