环境:
ubuntu18.04
GPU:GTX1060
显卡驱动版本:nvidia- driver- 470
pytorch 1.5.1 + torchvision0.6.0+ cuda10.2
参考链接:
搭建项目系统环境(CUDA+cuDNN+anaconda+pyCharm)
1.安装显卡驱动和CUDA
1.1安装显卡驱动
先更新源
sudo apt update
安装ubuntu-drivers,我们可以通过ubuntu-drivers检测你的NVIDIA显卡型号和推荐的驱动程序的模型。
sudo apt install ubuntu-drivers-common
获得NVIDIA显卡型号和推荐的驱动程序
ubuntu-drivers devices
我的系统推荐的显卡驱动版本 nvidia-driver-470,使用下面代码安装
sudo apt install nvidia-driver-470
Nvidia驱动默认安装OpenGL桌面,然而这又与GNOME桌面冲突,为了系统不宕机,需禁用nvidia的OpenGL。Linux系统一般默认安装的是开源的nouvea显卡驱动,它与nvidia显卡驱动产生冲突,欲装nvidia必禁nouvea!
禁用nouveau驱动:
sudo gedit /etc/modprobe.d/blacklist.conf
在弹窗最后加入下面代码:
sudo gedit /etc/modprobe.d/blacklist.conf
保存,关闭弹窗;至此,安装完毕,重启
sudo reboot
打开终端检查是否禁用成功,执行下面命令若无输出代表禁用nouveau生效
lsmod | grep nouveau
测试安装是否成功,这会返回你的所有显卡的信息
nvidia-smi
1.2下载完整版cuda
我选择CUDA10.2版本,进入nvidia的CUDA驱动下载页:
https://developer.nvidia.com/cuda-downloads


通过指令下载
wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2

本文详细介绍了如何在Ubuntu18.04系统上为GTX1060显卡安装nvidia-driver-470,搭配CUDA10.2、cudnn,并配置PyCharm开发环境,以便于实现如PVNet位姿估计等深度学习项目的复现。
搭建项目系统环境(CUDA+cuDNN+anaconda+pyCharm)&spm=1001.2101.3001.5002&articleId=137449808&d=1&t=3&u=e5d8fdb3810e4b1b9f97dc6d5fe0fb50)
4050

被折叠的 条评论
为什么被折叠?



