Python 安装
- Anaconda 安装(注:安装进度完成后,勾选自动添加路径)
- 创建环境并激活环境
conda create -n Gtensorflow pip python=3.5
注:尽量使用pyhton3.5
activate Gtensorlow
注:可以不创建环境,直接在base环境中安装 - 使用pip安装Tensorlow GPU版
pip install --ignore-installed --upgrade tensorflow-gpu==1.9 -i https://pypi.tuna.tsinghua.edu.cn/simple
注:==1.9
如果不写会默认安装低版本(1.1);-i
使用清华源安装
CUDA 安装
- 打开文件
Anaconda3\envs\tensorflow\Lib\site-packages\tensorflow\python\platform\build_info.py
看需要的版本
cudart_dll_name = 'cudart64_90.dll'
cuda_version_number = '9.0'
nvcuda_dll_name = 'nvcuda.dll'
cudnn_dll_name = 'cudnn64_7.dll'
cudnn_version_number = '7'
nvcuda.dll
和msvcp140.dll
在C:\Windows\System32\
中需要存在
cuda_version_number = '9.0'
和 cudnn_version_number = '7'
表示需要cuda9.0版本,cudnn7.0+版本
注:也可以查看TensorFlow官网:https://www.tensorflow.org/install/install_windows#requirements_to_run_tensorflow_with_gpu_support
- 在英伟达官网找对应版本的CUDA 和 cuDNN
CUDA:https://developer.nvidia.com/cuda-toolkit-archive
注意:win10 下需要下载主文件(1.4G)+多个补丁
cuDNN:https://developer.nvidia.com/rdp/cudnn-archive
-
安装CUDA
注意:(1)第一个大选项中,去掉勾选Visual Studio Integration
,(2)去掉Driver components
大选项(如果已经安装了N卡驱动,一般都安装了)(3)默认路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\
-
依次安装多个补丁包
注:会存在补丁包打开后闪退,可忽略 -
解压
cudnn-9.0-windows10-x64-v7.4.1.5.zip
,把bin,include,lib
里面的文件(共三个)分别拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\
下的bin,include,lib
相应目录中 -
在系统环境变量中
PATH
加入
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\libnvvp
测试
查看是否使用GPU
import tensorflow as tf
tf.test.gpu_device_name()
网友评论