(转载请注明出处)
电脑环境:windows10 + GeForce 840M + cuda9 + cudnn7 + tensorflow1.9+python3.5
1、官网直接下载anaconda
2、创建一个tensorflow虚拟环境,此处命名为tensorflow-gpu(或者可以直接在命令行创建,命令如下)
conda create -n tensorflow-gpu python==3.5
3、nVidia官网直接下载cuda9,安装,并配置环境变量
4、官网下载cudnn7,需要注册英伟达账号,注册时间比较慢,可以在网上另找资源,下载完成后,将文件解压,并按照文件夹名称分别放在cuda安装目录下,这里提供win10的cudnn7百度云链接,
链接:https://pan.baidu.com/s/15Z4wcHNid9KBNu9KTCu_EQ
提取码:ytah
5、进入cmd,命令
activate tensorflow-gpu
激活环境,紧接着执行以下命令来在线安装tensorflowGPU1.9
pip install tensorflow-gpu==1.9
6、pycharm配置python interpreter
7、测试代码为自己实验代码。不展示详细,贴出控制台打印出的显卡信息,证明测试成功(显示显卡信息)