Win10,GTX1050ti,Anaconda5.2,CUDA9.2,CUDNN7.1,VS2017安装tensorflow-gpu1.8

  • 本文的所有资料下载地址:链接:https://pan.baidu.com/s/1ymTc_eTZRDi403Qd0Tq3Wg 
    提取码:nmqn

1.本机配置:  win10 64位 + NVIDIA GeForce GTX 1050ti

2.安装VS2017,版本是15.6.7社区版

安装VS时,只勾选:通用Windows平台开发,使用C++的桌面开发

3.安装CUDA9.2.148

先安装cuda_9.2.148_win10.exe,

安装程序选择自定义安装(我之前选了精简安装,后面没装成功,连自己本身电脑的显卡驱动都坏了,重新下显卡驱动也装不上,还好我之前对c盘有过备份),我们只勾选CUDA里的Development、Runtime、Documentation、Samples。不要勾选Visual Studio Integration。按理说不勾选Visual Studio Integration之后是能顺利完成的。

    安装完毕后,我们打开cuda_9.2.88_win10.exe解压后cuda_9.2.88_win10的文件夹,进入到\cuda_9.2.88_win10\CUDAVisualStudioIntegration\extras\visual_studio_integration\MSBuildExtensions路径下,有四个文件,将所有文件拷贝到VS的对应目录下,参考路径如下
C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\Common7\IDE\VC\VCTargets\BuildCustomizations
然后我们双击下载的另一个exe文件:cuda_9.2.88.1_windows.exe,这应该是一个补丁文件,修复cuda_9.2.88_win10.exe中的一些问题的。

4.安装CUDNN

 解压cudnn-9.2-windows10-x64-v7.1.zip,将文件夹里的内容拷贝到CUDA的安装目录并覆盖相应的文件夹,CUDA拷贝目录:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2

5.Anaconda安装tensorflow-gpu1.8.0

安装Anaconda完之后,在Anaconda Prompt中输入conda create –n tensorflow python=3.6,然后自动创建一个tensorlfow的环境,创建完毕之后,activate tensorflow激活环境,然后用pip安装这个tensorlfow-gpu的whl文件,需要连网,因为会有其他的相关的库会自动下载

6.验证tensorflow(在Anaconda中tensorflow环境下的Spyder中验证)

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

输出:b'Hello, TensorFlow!'

7.Pycharm配置环境

打开Pycharm中的settings,然后选择Project Interpreter添加编译器

完毕,参考博文:https://blog.csdn.net/weixin_42359147/article/details/80622306#comments

此博文仅记录安装tensorflow-gpu的过程,以备以后再次使用

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值