ubuntu20.04+cuda11.8+cudnn8.9.6+pytorch2.4.1 + tensorflow2.12安装记录

本机环境

rtx4090
ubuntu20.04+cuda11.8+cudnn8.9.6安装记录

一:nvidia driver 安装

1.1 输入显卡型号查看支持显卡驱动的版本

进入英伟达中国驱动官网搜索支持的驱动版本
搜索支持的驱动在这里插入图片描述

1.2下载驱动并安装

方法一:直接在Ubuntu系统中的soft updates中安装535-server(专有)版本
在这里插入图片描述其他方法参考:因为可以直接使用方法一,避免麻烦就没测试其他方式,可自行参考
在ubuntu安装nvidia驱动 (亲测有效,这是方法二)
在ubuntu安装nvidia驱动(亲测有效,这是方法一))

二cuda11.8安装

安装过程中参考资料

1: Ubuntu 20.04 RTX 4090显卡 深度学习环境配置(Nvidia显卡驱动、CUDA11.6.0、cuDNN8.5)
2:揭秘Ubuntu深度学习服务器配置:新手如何成为专家?

2.1 安装cuda前的版本确定准备工作

进入官网
选择要安装的型号
【CUDA】Tensorflow、CUDA、cuDNN、python版本对应关系一览表(全)【win、macOS、linux】
这里是选择兼容pytorch和tensorflow的cudn版本,看个人需求

pytorch 应该是只要求cuda版本,对cudnn版本没有具体要求,而tensorflow对cuda cudnn版本都有要求
在这里插入图片描述pytorch cuda 版本对应参考资料感觉没有最新版本的对应关系,从官网可以看出支持关系:当前为最新支持版本所以选择11.8
此外还要注意当前安

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值