确定CUDA和cudnn版本

这里我是要用TensorFlow这个python库运行机器学习代码,因此要根据TensorFlow来配置CUDA和cudnn

查看TensorFlow官网,选择合适的版本方案

TensorFlow与CUDA版本对应图

安装CUDA

打开NVDIA控制面板,点击左下角系统信息,然后点击组件

NVDIA控制面板

这里的12.0是指cuda的最高版本,较低的版本也可以用,具体需要上网搜索和尝试。

NVDIA控制面板 系统信息

登录官网,下载合适的CUDA

官网下载地址

CUDA下载

下载完后点击安装即可,选择精简安装一路next

安装cudnn

官网下载cudnn

选择与CUDA对应的cudnn版本下载

将下载好的cudnn中的文件解压到cuda的安装目录下,GPU环境就配置好了

接下来只需在conda环境中下载相应的TensorFlow版本就可以使用GPU运行机器学习程序了

pytorch使用GPU

官方网站如下图页面,运行显示的命令能够更好的适配你的GPU

5

❤️ 转载文章请注明出处,谢谢!❤️