https://blog.csdn.net/insanegtp/article/details/108461944
如果您是高校学生或者高级研究人员,并且实验室或者个人资金充沛,建议您采用如下配置:
二、显卡安装
deepin20上的NVIDIA 显卡安装详见其它文章
三、CUDA10.1安装
1、查看最新或合适的tensorflow版本及其对应的各驱动组件匹配情况(显卡驱动、cuda、cudnn等):
https://tensorflow.google.cn/install/gpu?hl=en
2、在NVIDIA官网查看cuda和显卡驱动版本对应关系(https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html):
(如果做学习研究用,最好提前看好自己所用架构是否能支持所选cuda版本)
3、Cuda下载地址 (https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html)单击左侧Cuda版本,跳转后依次选择操作系统-计算机位数-系统型号-系统版本(选ubuntu版,cuda11可选择debian版)
4、安装,最终参考官网吧,一步一步来安装说明
(不同版本官网说明不同,自行查找)
查看GPU信息:lspci | grep -i nvidia
查看系统信息:uname -m && cat /etc/*release
查看GCC信息:gcc --version
打开终端,在数据盘新建cuda安装文件夹
mkdir /media/xingzhi/BigData/apps/cuda_10.1
进入cuda安装文件所在目录
cd /media/xingzhi/MyFiles/software/nvidia
直接运行下载的.run文件安装即可(该步很重要,需谨慎,否则容易报错)
sudo ./cuda_10.1.105_418.39_linux --silent --toolkit --toolkitpath=/media/xingzhi/BigData/apps/cuda_10.1 --defaultroot=/media/xingzhi/BigData/apps/cuda_10.1 --samples --samplespath=/media/xingzhi/BigData/apps/cuda_10.1
继续
输入accept
回车
取消掉Driver的勾选,安装其他的,Install回车,稍等安装结果如下(类似):
建立软连接
sudo ln -s /usr/local/cuda_10.1 /usr/local
添加环境变量
sudo vim /etc/profile
底部加入:
export CUDA_HOME=/usr/local/cuda_10.1
export PATH=/usr/local/cuda_10.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda_10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
然后source环境变量并重启系统
source /etc/profile
reboot
原文:https://www.cnblogs.com/lantingg/p/14675408.html