ubuntu 安装cuda及cudnn

进入nvidia开发者网站的CUDA下载页面:

CUDA Toolkit Archive | NVIDIA Developer

选择runfile格式的CUDA文件下载,下载完成后,解压,并运行上图中的命令,会有条款,接受即可,注意安装CUDA的时候不要安装驱动(因为在第一步我们已经安装过了)。

然后,

sudo vim ~/.bashrc

我们在文件最后一行添加:

export PATH="/usr/local/cuda-11.7/bin:$PATH"
export LD_LIBRARY_PATH="/usr/lcoal/cuda-11.7/lib64:$LD_LIBRARY_PATH"

最后,使刚才的配置生效。

source ~/.bashrc

进入/usr/local/cuda-10.1/samples/1_Utilities/目录,测试:

sudo make
cd deviceQuery
./deviceQuery

出现如下输出,则CUDA安装成功。

安装cuDNN

进入下载页面

cuDNN Archive | NVIDIA Developer

    sudo dpkg -i sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
    sudo dpkg -i sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
    sudo dpkg -i sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
sudo dpkg -l | grep cudnn


 

猜你喜欢

转载自blog.csdn.net/wang1625582254/article/details/131244202