深度学习环境配置:Tensorflow-gpu版本配置(含CUDA、cudnn安装配置)
1.查看你的电脑的显卡配置情况
查看显卡型号,是否支持CUDA
2.查看tf-gpu、cuda、cudnn版本对应关系
根据你所需的tensorflow-gpu型号,到下列网站查看相对应的cuda与cudnn版本
3.安装CUDA
可到官网直接下载,如果太慢,可国内网站找相应exe安装包
安装过程中出现警告:
安装过程可选择自定义,除主cuda程序外,另外两个驱动如果已经有更高版本可以不勾选安装
cuda主程序中的vs集成套件可不勾选安装
配置环境变量:
参考:
检查是否安装成功:
在命令行输入:
原创
2022-07-05 11:55:17 ·
767 阅读 ·
0 评论