安装 Ubuntu20.04安装 Cuda11和cuDNN
Cuda是Nvidia显卡计算环境,cuDNN是在Cuda基础上的深度学习库。
1) 具有足够空间(10GB以上)的Ubuntu系统;
2) 支持Cuda的显卡,此处为 GTX 1650, 1000元左右。
3) Internet网络连接,或提前下载。
可选系统自动安装和手动下载安装。
在https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html 这里可以查询到各个版本支持的情况。
在Ubuntu的“软件和更新/附加驱动”中,选择使用NVIDIA driver,而不使用系统自带的nouveau。
“应用更改”后系统下载驱动并安装。
在以下网址下载Nvidia驱动,选择自己显卡的型号。
https://www.nvidia.cn/geforce/drivers/ |
需要转入完全的命令窗口中,卸载原来的驱动,然后再安装,最后加入路径。
整个过程较麻烦。请自行百度。
- $ sudo nvidia-smi
- $ sudo nvidia-settings
在如下网址选择合适的安装包
https://developer.nvidia.com/cuda-downloads |
比如,本次配置为linux、x86_64、ubuntu、20.04、runfile(local),得到如下安装提示。
文件较大(3GB),网速很快,10分钟不到就下载完了。
sudo sh cuda_11.1.0_455.23.05_linux.run |
默认设置即可。
注意:前面已经安装驱动了,因此这一步不需要安装驱动了,否则会报错。
可能的错误:
如果出现gcc报错,则需要安装gcc。命令如下:
sudo apt-get install gcc |
在如下网址下载cuDNN,当前需要注册和填写一个简答的问卷调查。
https://developer.nvidia.com/cudnn-download-survey |
选择正确的版本,文件有好几个需要下载,有运行环境,开发环境和示例。
下载的Deb文件,直接双击即可安装。
根据提示,将路径加入系统PATH中。
添加路径方法.
测试CUDA的sammples,主要一本机为例,执行如下代码,有结果则表明安装正确。
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
#由自己电脑目录决定 make sudo ./deviceQuery |
尚无评论!