本机环境
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
此外还要注意当前安