TensorFlow——GPU版本安装
参考文章 1.http://blog.sina.com.cn/s/blog_14935c5880102wu86.html
2.https://www.cnblogs.com/hear-nothing/p/7464882.html 镜像下载
3.https://blog.youkuaiyun.com/gangeqian2/article/details/79358543 安装
1.winR打开诊断看电脑GPU信息dxdiag
查看电脑GPU版本 https://developer.nvidia.com/cuda-gpus
2.打开NVIDIA控制面板->帮助->组件查看当前显卡支持CUDA Tool kit的版本
CPU下训练模型很慢,GPU则相对快速很多,NVIDIA有GPU,可以安装,用GPU在代码中加入两行,0代表第一块GPU设备
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0"
(1)安装CUDA
可看做GPU的驱动
1.准备好NVIDIA的显卡,下载安装CUDA
https://developer.nvidia.com/cuda-toolkit-archive
下载完后,双击就可安装,next到底
这样就全部安装成功。
验证安装成功:打开cmd,输入nvcc -V,显示如下,则CUDA安装成功。
2.安装好之后把CUDA安装目录下的bin和lib\x64添加到Path环境变量中
安装目录:(1)若为