Windows:深度学习基础GPU环境配置(tensorflow,keras,caffe)

Step1:下载并安装Nvidia driver

下载地址

Step2:下载并安装CUDA

下载地址
(1)需要编译caffe的的朋友,注意选择自定义安装并勾选visual studio integration
(2)RTX20系列显卡,RTX16系列显卡最好CUDA10.0以上版本
(3)如搭配tensorflow等,需要自行按照官网匹配相应的CUDA版本

Step3:下载cudnn

下载地址
下载后直接解压打开,然后将bin,include,lib文件夹直接剪切粘贴到
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1文件夹下即可

Step4:确认安装完成

开CMD命令行,输入nvcc -V,如果未报错,则证明安装完成

Step5:下载安装Anaconda

下载地址

Step6:配置keras+tensorflow训练环境

windows10 按 win + q键,搜索anaconda prompt,打开依次输入以下命令行

Conda create --name python3 python=3.7
Conda activate python3
Conda install keras-gpu
Conda install tensorflow=1.14 (此行一般不需要)

linux同理,自行找到anaconda prompt命令提示符,然后输入上述命令即可

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值