目录
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命令提示符,然后输入上述命令即可