Ubuntu使用GPU深度学习加速的环境配置
1. 下载GPU驱动去官网根据条件下载对应的GPU驱动文件,得到.run文件2. 禁用Nouveau 驱动禁用 Nouveau 驱动3. 安装驱动依次执行以下命令:sudo apt install makesudo apt install gccsudo chmod 777 xx.runsudo apt-get remove --purge nvidia*sudo bash xx.run然后出现选
·
1. 下载GPU驱动
去官网根据条件下载对应的GPU驱动文件,得到.run文件
2. 禁用Nouveau 驱动
3. 安装驱动(选择这一步,则在安装CUDA时可以不选择安装驱动;否则就需要选择安装CUDA时的驱动)
依次执行以下命令:
sudo apt install make
sudo apt install gcc
sudo chmod 777 xx.run
sudo apt-get remove --purge nvidia*
sudo bash xx.run
然后出现选择时,一路默认即可
最后验证是否安装成功,使用命令nvidia-smi
,会出现以下证明成功了:
4. 安装CUDA
去官网下载相应版本的CUDA文件,文件格式为.run
在选择安装内容时,可以将驱动模块
取消
sudo chmod 777 xx.run
sudo bash xx.run
安装完成后会出现如下信息:
意味着我们需要配置环境变量,配置环境变量相关教程参考:添加cuda到bashrc环境变量
安装成功后,使用命令nvcc -V
查看CUDA版本:
5. 安装CUDNN
5.1 若是deb文件
官网下载对应版本的CUDNN,我下载的是deb版本,然后使用命令安装:sudo dpkg -i xxx.deb
即可安装成功
- 注意:如果使用Paddle-GPU版本,可能会报错如下:
这是因为cudnn虽然安装了, 但是没有指向,导致paddle找不到,那么参考这个教程:百度Paddlepaddle-GPU运行时出现Error: Cannot load cudnn shared library. Cannot invoke method cudnnGetVersion,只需要将libcudnn.so
文件连接好即可,另一个so可不管
更多推荐
已为社区贡献6条内容
所有评论(0)