Python3.7+Pycharm+cuda10.0+tensorflow GPU版本 安装

处理器:I5-7500

显卡 :GTX1050Ti

系统 :Win10

1. 首先搭建Python环境。

官网https://www.python.org/downloads/下载Python3.7(目前最新的版本为3.7.4,暂时不用最新版本)。下载安装完成后在环境变量中添加bin路径。

cmd 运行 python 查看是否配置成功。

C:\Users\admin>python
Python 3.7.0 (v3.7.0:1bf9cc5093, Jun 27 2018, 04:59:51) [MSC v.1914 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>>

2. 下载Cuda

官网链接:https://developer.nvidia.com/cuda-downloads

网盘链接: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-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:04_Central_Daylight_Time_2018
Cuda compilation tools, release 10.0, V10.0.130

4.tensorflow安装

1> 通过pip install --upgrade tensorflow-gpu 安装,验证是否成功。

import tensorflow as tf
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。