300字范文,内容丰富有趣,生活中的好帮手!
300字范文 > tensorflow GPU环境配置 Nvidia+cuda+cudnn

tensorflow GPU环境配置 Nvidia+cuda+cudnn

时间:2021-12-25 13:10:03

相关推荐

tensorflow GPU环境配置 Nvidia+cuda+cudnn

显卡安装

去官网下载显卡驱动/

删除旧的nvidia驱动(刚装的机子不需要这一步)

sudo apt-get purge nvidia*

sudo chmod +x *.run

sudo ./NVIDIA-Linux-x86_64-430.34.run --uninstall

禁用自带的nouveau nvidia驱动sudo vim /etc/modprobe.d/blacklist-nouveau.conf

并添加如下内容:

blacklist nouveauoptions nouveau modeset=0

更新驱动配置

sudo update-initramfs -u

重启

reboot

确认是否成功,如果没有显示内容则成功

lsmod | grep nouveau

3.切换到命令行模式

关闭界面程序sudo service lightdm stop切换到tty1控制台:Ctrl+Alt+F1

sudo chmod +x NVIDIA-Linux-x86_64-430.34.run

sudo ./NVIDIA-Linux-x86_64-430.34.run

很多博客说要加上== -no-opengl-files==参数,实测这样在运行图形sample示例的时候会报错,所以这里我没加

测试是否安装成功

nvidia-smi

sudo service lightdm start

cuda

1.确认有支持cuda的gpu

lspci | grep -i nvidia

/cuda-gpus查看该gpu是否支持

2.查看linux内核是否支持

uname -m && cat /etc/*release

3.是有有gcc

gcc --version

4.确认系统头

uname -r

5.选择一种安装方式(这一步只是介绍安装方法,并没有操作什么)

一种是deb一种是run.推荐deb安装

6.下载cuda,并确认

md5sum <file>

7.安装之前,要把之前安装的cuda卸载掉

8.正式安装

sh ./cuda_10.1.168_418.67_linux.run

不要选择安装驱动,因为之前已经安装过

9.配置环境

这两句话加载./.bashrc

export PATH=/usr/local/cuda-10.1/bin:/usr/local/cuda-10.1/NsightCompute-.1${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

第二句第一行的斜杠是转义,不是换行,如果和在一行要带上.

检验

nvcc -V查看是否安装成功

cudnn安装

mv cudnn-10.1-linux-x64-v7.6.2.24.solitairetheme8 cudnn-10.1-linux-x64-v7.6.2.24.tgztar -xzvf cudnn-10.1-linux-x64-v7.6.2.24.tgzsudo cp cuda/include/cudnn.h /usr/local/cuda/includesudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

/deeplearning/sdk/cudnn-install/index.html

参考文献:

/cuda/cuda-installation-guide-linux/index.html

/luofeel/p/8654964.html很多博客说要加上== -no-opengl-files参数,实测这样在运行图形sample示例的时候会报错,所以这里我没加

/u014682691/article/details/80605201很多博客说要加上-no-opengl-files==参数,实测这样在运行图形sample示例的时候会报错,所以这里我没加

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。