保姆级教程:如何让30系显卡用上 Tensorflow 1.15.4

2021-01-06 22:55:43 31点赞 73收藏 12评论

你是AMD Yes党?还是intelNVIDIA的忠实簇拥呢?最新一届#装机大师赛#开始啦!本次装机阵营赛分为3A红组intel NVIDIA蓝绿组混搭组还有ITX组,实体or虚拟装机都能参与,可使用值得买定制化DIY装机工具在文中展现配置单!每个小组均有精美礼品,优秀文章还可角逐装机大师终极大奖,点击参与<<<

创作立场声明:原创

前言

相较于20系显卡,30系显卡性能的提升非常大。因此近期本人入手了一块3090。

微星(MSI)超龙GeForceRTX3090SUPRIMX24G超旗舰电竞游戏设计智能学习电脑独立显卡15999元京东去购买

然而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). 如果运行如下指令没用打印出任何内容, 则禁用nouveau成功.

lsmod | grep nouveau


4.关闭用户图形界面,并安装zhcon(如果出现中文乱码)

sudo systemctl set-default multi-user.target

sudo apt-get install zhcon

sudo adduser $(whoami) video

sudo reboot



5.安装显卡驱动和cuda11.1

三种安装方式:

a.runfile(local) b.deb(local) c.deb(network)

保姆级教程:如何让30系显卡用上 Tensorflow 1.15.4

推荐用runfile方式,不管之前有没有装驱动,这里先卸载原有驱动

sudo zhcon --utf8

sudo apt-get remove --purge nvidia*

sudo apt-get update

sudo apt-get install dkms build-essential linux-headers-generic

运行安装文件

sudo chmod a+x cuda_11.1.0_455.23.05_linux.run

sudo ./cuda_11.1.0_455.23.05_linux.run

运行以后会看到协议页面,输入 accept

保姆级教程:如何让30系显卡用上 Tensorflow 1.15.4

然后进入安装选项页面,这里要关闭OpenGL!!!!

Options - Driver Options- Do not install any of the OpenGL-related driver files

保姆级教程:如何让30系显卡用上 Tensorflow 1.15.4

保姆级教程:如何让30系显卡用上 Tensorflow 1.15.4

保姆级教程:如何让30系显卡用上 Tensorflow 1.15.4

然后选择 Done 退回到初始页面,选择 Install 开始安装。安装完成后开启用户图形界面。

sudo systemctl set-default graphical.target

sudo reboot

查看NVIDIA显卡驱动是否安装完成 nvidia-smi


6.配置路径(这里还包括anaconda3,texlive,matlab)

打开bashrc

sudo gedit ~/.bashrc


在最后添加路径

(这里的cuda路径不加版本号,以便后续升级cuda版本的时候不需要重新配置)

(这里还包含了anaconda3,texlive,matlab2020b)

export CUDA_HOME=/usr/local/cuda

export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:${CUDA_HOME}/lib64

export PATH=${CUDA_HOME}/bin:${PATH}


export PATH="/home/用户名/anaconda3/install/bin:$PATH"


export MANPATH=${MANPATH}:/usr/local/texlive/2020/texmf-dist/doc/man

export INFOPATH=${INFOPATH}:/usr/local/texlive/2020/texmf-dist/doc/info

export PATH=${PATH}:/usr/local/texlive/2020/bin/x86_64-linux


export PATH=/usr/local/Polyspace/R2020b/bin:$PATH


保存,退出,输入 source ~/.bashrc 使其生效。


验证CUDA版本,会输出版本号

nvcc --version

cat /proc/driver/nvidia/version


7.安装cudnn 8.0.4

下载cuDNN Library for Linux(x86_64)并解压,在文件所在的路径下打开终端

sudo cp cuda/lib64/* /usr/local/cuda/lib64/

sudo cp cuda/include/* /usr/local/cuda/include/

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*


最后验证cudnn是否配置好并输出版本号:

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2


8.安装anaconda3,创建并进入虚拟环境

conda create -n name python=3.6.8

conda activate name


9.安装tensorflow

运行requirements.sh安装安装nvidia-TensorFlow对应的依赖包:

sh requirements.sh

验证库文件

pip list | grep nvidia

保姆级教程:如何让30系显卡用上 Tensorflow 1.15.4

测试Tensorflow是否安完成

python -c 'import tensorflow as tf; print(tf.__version__)'

保姆级教程:如何让30系显卡用上 Tensorflow 1.15.4


总结

测试了一下,Tensorflow 1.15.4可以向下兼容到 Tensorflow1.13.1版本,再往下就没测试。


展开 收起

MSI 微星 万图师 GeForce RTX 3070 VENTUS 3X 8G OC 超频版 三风扇 电脑显卡

MSI 微星 万图师 GeForce RTX 3070 VENTUS 3X 8G OC 超频版 三风扇 电脑显卡

4599元起

MSI 微星 万图师 GeForce GTX 1660 VENTUS XS C 6G OCV1 显卡 6GB

MSI 微星 万图师 GeForce GTX 1660 VENTUS XS C 6G OCV1 显卡 6GB

1599元起

MSI 微星 GeForce RTX 3060 Ti GAMING X TRIO 显卡 8GB

MSI 微星 GeForce RTX 3060 Ti GAMING X TRIO 显卡 8GB

3999元起

MSI 微星 魔龙 GeForce RTX 3070 GAMING X TRIO 8G 超频版 旗舰款 电脑显卡

MSI 微星 魔龙 GeForce RTX 3070 GAMING X TRIO 8G 超频版 旗舰款 电脑显卡

4999元起

MSI 微星 魔龙 GeForce RTX3090 GAMING X TRIO 24G 显卡 24GB

MSI 微星 魔龙 GeForce RTX3090 GAMING X TRIO 24G 显卡 24GB

13999元起

MSI 微星 旗舰魔龙 GeForce GTX 1660 SUPER GAMING X 显卡 6GB

MSI 微星 旗舰魔龙 GeForce GTX 1660 SUPER GAMING X 显卡 6GB

2299元起

MSI 微星 GeForce RTX 2070 SUPER GAMING X TRIO 显卡 8GB

MSI 微星 GeForce RTX 2070 SUPER GAMING X TRIO 显卡 8GB

3799元起

MSI 微星 超龙 GeForce RTX 3080 SUPRIM X 10G 超旗舰

MSI 微星 超龙 GeForce RTX 3080 SUPRIM X 10G 超旗舰

7499元起

MSI 微星 魔龙 GeForce GTX 1650 GAMING X 显卡 4GB

MSI 微星 魔龙 GeForce GTX 1650 GAMING X 显卡 4GB

1199元起

MSI 微星 GTX 1050 Ti-信仰红龙 GAMING 4GB

MSI 微星 GTX 1050 Ti-信仰红龙 GAMING 4GB

1099元起

微星(MSI)魔龙Z GeForce GTX 1660 SUPER GAMING Z PLUS 6G 1660S超旗舰 电竞游戏台式电脑独立显卡

微星(MSI)魔龙Z GeForce GTX 1660 SUPER GAMING Z PLUS 6G 1660S超旗舰 电竞游戏台式电脑独立显卡

2199元起

MSI 微星 魔龙 GeForce RTX 2060 GAMING Z 显卡 6GB

MSI 微星 魔龙 GeForce RTX 2060 GAMING Z 显卡 6GB

2999元起

微星(MSI)超龙 GeForce RTX 3090 SUPRIM X 24G 超旗舰 电竞游戏设计智能学习电脑独立显卡

微星(MSI)超龙 GeForce RTX 3090 SUPRIM X 24G 超旗舰 电竞游戏设计智能学习电脑独立显卡

15999元起

微星(MSI)GT 710 1GD3H LP 迷你半高静音刀卡 台式电脑游戏办公设计设备独立显卡

微星(MSI)GT 710 1GD3H LP 迷你半高静音刀卡 台式电脑游戏办公设计设备独立显卡

299元起

微星(MSI)GT 710 1GD3H LP 迷你半高静音刀卡 台式电脑游戏办公设计设备独立显卡

微星(MSI)GT 710 1GD3H LP 迷你半高静音刀卡 台式电脑游戏办公设计设备独立显卡

259元起

MSI 微星 万图师 GeForce RTX 3090 VENTUS 3X 24G 显卡 24GB

MSI 微星 万图师 GeForce RTX 3090 VENTUS 3X 24G 显卡 24GB

11999元起

文中相关商品

京东 微星(MSI)超龙 GeForce RTX 3090 SUPRIM X 24G 超旗舰 电竞游戏设计智能学习电脑独立显卡
12评论

发表评论请 登录
  • 最新
  • 最热
评论举报

请选择举报理由

相关好价推荐
查看更多好价

相关文章推荐

更多精彩文章
更多精彩文章
最新文章 热门文章
73
扫一下,分享更方便,购买更轻松