zl程序教程

您现在的位置是:首页 >  其他

当前栏目

CUDA deb 安装

安装 CUDA deb
2023-06-13 09:11:57 时间

大家好,又见面了,我是你们的朋友全栈君。

1、CUDA下载

https://developer.nvidia.com/cuda-toolkit-archive

2、选择对应版本deb安装

sudo dpkg -i cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

3、添加环境变量

sudo gedit ~/.bashrc

后面添加

export PATH=/usr/local/cuda-10.0/bin${PATH:+:$PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

更新

sudo source ~/.bashrc

编译测试deviceQuery

$ cd /usr/local/cuda-10.0/samples/1_Utilities/deviceQuery
$ sudo make
$ ./deviceQuery

如果结果是 Result = PASS CUDA 安装成功

4、cndnn下载,安装libcudnn7、libcudnn7-dev,注意与CUDA版安装

https://developer.nvidia.com/rdp/cudnn-download

libcudnn7_7.5.0.56-1+cuda10.0_amd64.deb

libcudnn7-dev_7.5.0.56-1+cuda10.0_amd64

发布者:全栈程序员栈长,转载请注明出处:https://javaforall.cn/135972.html原文链接:https://javaforall.cn