TensorFlow——GPU安装

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)若为

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值