Linux/Ubuntu系统如何安装cudnn?(适用于所有40系显卡 4090 4080 4070)

您所在的位置:网站首页 如何装ubuntu系统 Linux/Ubuntu系统如何安装cudnn?(适用于所有40系显卡 4090 4080 4070)

Linux/Ubuntu系统如何安装cudnn?(适用于所有40系显卡 4090 4080 4070)

2023-04-02 08:24| 来源: 网络整理| 查看: 265

下载安装CUDA

卸载掉原有驱动:sudo apt-get remove nvidia-* 删除旧内核:sudo apt autoremove 安装NVIDIA显卡驱动:sudo apt install nvidia-driver-525 官网下载:https://developer.nvidia.com/cuda-toolkit-archive

下载安装CUDNN

官网下载:https://developer.nvidia.com/rdp/cudnn-download 自行选择与您下载的CUDA对应的适合的版本

加入环境变量

解压命令:tar -xf cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive.tar.xz注意,名字替换为你下载的版本 进入到cudnn的下载目录,然后输入以下代码,将解压后的文件拷贝到CUDA对应的安装目录下,并更改文件属性: 1、sudo cp include/cudnn.h /usr/local/cuda/include 2、sudo cp lib/libcudnn* /usr/local/cuda/lib64*

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

把cuda文件夹下的bin目录添加到path中 5 export PATH=$PATH:/usr/local/cuda/bin 6 export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

要使环境变量修改生效,命令行输入:source ~/.bashrc

可以在终端上查看路径有没有添加进去 echo $PATH echo $LD_LIBRARY_PATH

查看cuda有没有安装好: 到安装的cuda的bin目录下,打开终端输入命令:nvcc -V

然后sudo reboot 重启即可~



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3