form1.cn
Make a little progress every day

Ubuntu 16.04. 装tesla p4 显卡驱动+cuda9.0

27th of March 2022 Linux Command 63

显卡和cuda都要重新安装一次才可以


显卡下载地址:https://www.nvidia.cn/Download/index.aspx?lang=cn

cuda、显卡版本和tensorflow-gpu对应关系:https://blog.csdn.net/weixin_45284801/article/details/105920249


要根据显卡确定要安装哪个cuda

cuda 9.0地址:

https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal


根据cuda 版本 确定 加速库 cudnn 的版本 ,cudnn不是必须装的,但是装了可以提升1.5倍速度

cuDNN(CUDA Deep Neural Network library):是NVIDIA打造的针对深度神经网络的加速库,是一个用于深层神经网络的GPU加速库。如果你要用GPU训练模型,cuDNN不是必须的,但是一般会采用这个加速库。


卸载cuda和显卡驱动

cuda: /cuda/bin/uninstall_cuda_9.0.pl

显卡:/usr/bin/nvidia-uninstall


配置CUDA环境变量

sudo vim ~/.bashrc

在末尾加入:

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}

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

export CUDA_HOME=/usr/local/cuda


source ~/.bashrc


参考链接:

https://blog.csdn.net/qq_25387073/article/details/99623872

下一篇: 没有了