前言
相较于20系显卡,30系显卡性能的提升非常大。因此近期本人入手了一块3090。然而30系显卡目前只支持CUDA11以上的版本。虽然Tensorflow和Pytorch最新版本已经适配了CUDA11.1,但是本人还有一些代码需要运行在Tensorflow 1.X版本上。网上已经有些教程了,然而个人感觉过程不太详细,所以这里总结一下从UBUNTU安装完成到Tensorflow 1.15.4配置完成的整个过程。
前提:必须使用Linux系统,目前没有研究出Windows的安装方法。推荐使用Ubuntu18.04。
过程
Ubuntu为刚安装完成的状态,没有进行任何设置。
配置环境所需要的安装包(ed2t )
1.下载cuda11.1和cudnn8.0.4,
可以自行在官网下载,也可以从上面的链接直接下载。记住下载目录,最好放在用户目录下(/home/用户名/),方便后面使用。
2.禁用nouveau
在终端输入
sudo gedit /etc/modprobe.d/blacklist.conf
在blacklist.conf的最后添加:
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off
保存并退出
3.禁用nouveau内核模块
echo options nouveau modeset=0
sudo update-initramfs -u
重启(sudo reboot). 如果运行如下指