处理器:I5-7500
显卡 :GTX1050Ti
系统 :Win10
1. 首先搭建Python环境。
官网https://www.python.org/downloads/下载Python3.7(目前最新的版本为3.7.4,暂时不用最新版本)。下载安装完成后在环境变量中添加bin路径。
cmd 运行 python 查看是否配置成功。
C:\Users\admin>python
Python3.7.0 (v3.7.0:1bf9cc5093, Jun 27 2018, 04:59:51) [MSC v.1914 64bit (AMD64)] on win32
Type"help", "copyright", "credits" or "license" formore information.>>>
2. 下载Cuda
网盘链接:https://pan.baidu.com/s/1K_PEdQ-aJRu3mnCpVvPREQ
提取码:l2vw
按步骤安装。
3. Cuddn
链接:https://pan.baidu.com/s/1BXCwyUb6fUF7ULfKJd_8VA
提取码:qlgy
将解压出来的文件替换到CUDA\v10.0\bin目录下。
cmd 运行 nvcc -V 查看是否安装成功。
C:\Users\admin>nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c)2005-2018NVIDIA Corporation
Built on Sat_Aug_25_21:08:04_Central_Daylight_Time_2018
Cuda compilation tools, release10.0, V10.0.130
4.tensorflow安装
1> 通过pip install --upgrade tensorflow-gpu 安装,验证是否成功。
import tensorflow astf
hello= tf.constant('Hello, TensorFlow!')
sess=tf.Session()
print(sess.run(hello))
2>打开pycharm,在菜单栏里Flie-Settings-Project-Project interpreter,选择Python3 interpreter,然后点击+按钮,输入tensorflow-gpu,install package。